https://www.tensorflow.org/agents/api_docs/python/tf_agents/metrics/py_metric/PyStepMetric?authuser=1
tf_agents.metrics.py_metric.PyStepMetric | TensorFlow Agents
Defines the interface for metrics that operate on trajectories.
tfagentsmetricspytensorflow