Module: tf_agents.distributions.gumbel_softmax

View source on GitHub

Gumbel_Softmax distribution classes.

Classes

class GumbelSoftmax: GumbelSoftmax distribution with temperature and logits.