jaxopt.perturbations.Gumbel

class jaxopt.perturbations.Gumbel[source]

Gumbel distribution.

__init__()

Methods

log_prob(inputs)

rtype: Array

sample(seed, sample_shape)

rtype: Array