mxnet.gluon.probability.distributions.relaxed_one_hot_categorical¶

Relaxed Bernoulli class.

Classes

RelaxedOneHotCategorical(T, num_events[, ...])

Create a relaxed one hot categorical distribution object.

class mxnet.gluon.probability.distributions.relaxed_one_hot_categorical.RelaxedOneHotCategorical(T, num_events, prob=None, logit=None, validate_args=None)[source]¶

Bases: TransformedDistribution

Create a relaxed one hot categorical distribution object.

Parameters:

T (scalar, default None) – Relaxation temperature
num_events (Int) – Number of events.
prob (Tensor) – Probabilities of each event.
logit (Tensor) – The log-odds of each event