Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.specs.tensor_spec.add_outer_dims_nest

View source on GitHub

Adds outer dimensions to the shape of input specs.

tf_agents.specs.tensor_spec.add_outer_dims_nest(
    specs, outer_dims
)

Args:

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

Returns:

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

Raises:

  • ValueError: if any outer_dims is neither a list nor tuple.