ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf_agents.utils.composite.reshape

Reshape composite tensor t to shape.

t A Tensor or SparseTensor.
shape 1D tensor, array, or list. The new shape.

The reshaped tensor.