Module: tf_agents.distributions

Distributions module.

Modules

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-04-26 UTC.