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

tf_agents.utils.common.replicate

View source on GitHub

Replicates a tensor so as to match the given outer shape.

tf_agents.utils.common.replicate(
    tensor, outer_shape
)

Example:

  • t = [1, 2, 3], [4, 5, 6]
  • outer_shape = [2, 1] The shape of the resulting tensor is: [2, 1, 2, 3] and its content is: [[t], [t]]

Args:

  • tensor: A tf.Tensor.
  • outer_shape: Outer shape given as a 1D tensor of type list, numpy or tf.Tensor.

Returns:

The replicated tensor.

Raises:

  • ValueError: when the outer shape is incorrect.