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

tf_agents.utils.common.scale_to_spec

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.