View source on GitHub |
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.