Modify evaluate function for reporting extra metrics to support some internal downstream changes.
Modify evaluate function for reporting extra metrics to support some internal downstream changes.

PiperOrigin-RevId: 463902205
Change-Id: I7bd71fcf792e95bb192a391c85e86126c6ca7fdf

google-research/circuit_training