tf_agents.utils.composite.expand_dims

Stay organized with collections Save and categorize content based on your preferences.

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.