tfp.edward2.ExpRelaxedOneHotCategorical

Create a random variable for ExpRelaxedOneHotCategorical.

See ExpRelaxedOneHotCategorical for more details.

RandomVariable.

Original Docstring for Distribution

Initialize ExpRelaxedOneHotCategorical using class log-probabilities.

temperature A Tensor, representing the temperature of one or more distributions. The temperature values must be positive, and the shape must broadcast against (logits or probs)[..., 0].
logits An N-D Tensor, N >= 1, representing the log probabilities of one or many distributions. The first N - 1 dimensions index into a batch of independent distributions and the last dimension represents a vector of logits for each class. Only one of logits or probs should be passed in.
probs An N-D Tensor, N >= 1, representing the probabilities of one or many distributions. The first N - 1 dimensions index into a batch of independent distributions and the last dimension represents a vector of probabilities for each class. Only one of logits or probs should be passed in.
validate_args Python bool, default False. When True distribution parameters are checked for validity despite possibly degrading runtime performance. When False invalid inputs may silently render incorrect outputs.
allow_nan_stats Python bool, default True. When True, statistics (e.g., mean, mode, variance) use the value "NaN" to indicate the result is undefined. When False, an exception is raised if one or more of the statistic's batch members are undefined.
name Python str name prefixed to Ops created by this class.