View source on GitHub |
Converts TensorSpec into ArraySpec.
tf_agents.specs.tensor_spec.to_array_spec(
tensor_spec: Union[tf_agents.typing.types.ArraySpec
, tf_agents.typing.types.DistributionSpecV2
]
) -> tf_agents.typing.types.NestedArraySpec