View source on GitHub |
Shapes and scales a batch into the given spec bounds.
tf_agents.utils.common.scale_to_spec(
tensor, spec
)
Used in the notebooks
Used in the tutorials |
---|
Args | |
---|---|
tensor
|
A [batch x n] tensor with values in the range of [-1, 1]. |
spec
|
(BoundedTensorSpec) to use for scaling the action. |
Returns | |
---|---|
A batch scaled the given spec bounds. |