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

Module: tf_agents.distributions.gumbel_softmax

Gumbel_Softmax distribution classes.

Classes

class GumbelSoftmax: GumbelSoftmax distribution with temperature and logits.