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

tf_agents.utils.composite.expand_dims

View source on GitHub

Add a new dimension to tensor t along axis.

tf_agents.utils.composite.expand_dims(
    t, axis
)

Args:

Returns:

An expanded tensor.

Raises:

  • NotImplementedError: If t is a SparseTensor and axis != 0.