ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf_agents.distributions

Distributions module.


gumbel_softmax module: Gumbel_Softmax distribution classes.

masked module: Define distributions for spaces where not all actions are valid.

reparameterized_sampling module: Helper function to do reparameterized sampling if the distributions supports it.

shifted_categorical module: Categorical distribution where values are shifted to honor a range.

tanh_bijector_stable module: Tanh bijector.

utils module: Utilities related to distributions.