tf_agents.utils.composite.expand_dims

View source on GitHub

Add a new dimension to tensor t along axis.

t A tf.Tensor or tf.SparseTensor.
axis A 0D integer scalar.

An expanded tensor.

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