tf_agents.utils.composite.reshape

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

Reshape composite tensor t to shape.

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

The reshaped tensor.