tf_agents.utils.composite.reshape

View source on GitHub

Reshape composite tensor t to shape.

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

The reshaped tensor.