Module: tf_agents.distributions.utils

View source on GitHub

Utilities related to distributions.

Classes

class SquashToSpecNormal: Scales an input normalized action distribution to match spec bounds.

Functions

scale_distribution_to_spec(...): Scales the given distribution to the bounds of the given spec.