tf_agents.utils.common.replicate

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.