View source on GitHub |
Generates summaries of all tensors in tensor_dict
.
tf_agents.utils.common.summarize_tensor_dict(
tensor_dict: Dict[Text, tf_agents.typing.types.Tensor
],
step: Optional[types.Tensor]
)
Args | |
---|---|
tensor_dict
|
A dictionary {name, tensor} to summarize. |
step
|
The global step |