Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.utils.composite.reshape

View source on GitHub

Reshape composite tensor t to shape.

tf_agents.utils.composite.reshape(
    t, shape
)

Args:

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

Returns:

The reshaped tensor.