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

Module: tf_agents.distributions.gumbel_softmax

View source on GitHub

Gumbel_Softmax distribution classes.

Classes

class GumbelSoftmax: GumbelSoftmax distribution with temperature and logits.