ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf_agents.specs.tensor_spec.add_outer_dim

Adds an outer dimension to the shape of input specs.

Used in the notebooks

Used in the tutorials

specs Nested list/tuple/dict of TensorSpecs/ArraySpecs, describing the shape of tensors.
dim Int, representing the outer dimension to be added to the TensorSpecs in specs.

Nested TensorSpecs with outer dimensions added to the shape of input specs.