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