Implementation of the Gumbel-Sigmoid distribution in PyTorch.
pytorch gumbel-softmax gumbel-distribution reparameterization-trick straight-through-gradient gumbel-sigmoid
-
Updated
Jul 22, 2022 - Python