tf_agents.specs.tensor_spec.add_outer_dim

Adds an outer dimension to the shape of input specs.

tf_agents.specs.tensor_spec.add_outer_dim(
    specs, dim=None
)

Used in the notebooks

Used in the tutorials
Train a Deep Q Network with TF-Agents REINFORCE agent

Args
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.

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