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. |