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

tf_agents.utils.common.transpose_batch_time

View source on GitHub

Transposes the batch and time dimensions of a Tensor.

tf_agents.utils.common.transpose_batch_time(
    x
)

If the input tensor has rank < 2 it returns the original tensor. Retains as much of the static shape information as possible.

Args:

  • x: A Tensor.

Returns:

x transposed along the first two dimensions.