https://www.tensorflow.org/probability/oryx/api_docs/python/oryx/distributions/ExpRelaxedOneHotCategorical
oryx.distributions.ExpRelaxedOneHotCategorical | Oryx | TensorFlow
ExpRelaxedOneHotCategorical distribution with temperature and logits.
oryxdistributionstensorflow