tf_agents.specs.tensor_spec.add_outer_dims_nest

View source on GitHub

Adds outer dimensions to the shape of input specs.

specs Nested list/tuple/dict of TensorSpecs/ArraySpecs, describing the shape of tensors.
outer_dims a list or tuple, representing the outer shape to be added to the TensorSpecs in specs.

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

ValueError if any outer_dims is neither a list nor tuple.