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

Module: tf_agents.distributions

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.