Skip to content

Implementation of the Gumbel-Sigmoid distribution in PyTorch.

Notifications You must be signed in to change notification settings

AngelosNal/PyTorch-Gumbel-Sigmoid

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Gumbel-Sigmoid

This repository contains a PyTorch implementation of the Gumbel-Sigmoid distribution.

The code is adapted from the official PyTorch implementation of the Gumbel-Softmax distribution (link).

Example

In  [1]: import torch

In  [2]: from gumbel_sigmoid import gumbel_sigmoid

In  [3]: x = torch.rand(3,2)

In  [4]: gumbel_sigmoid(x, hard=False)
Out [4]: tensor([[0.3771, 0.4769],
                 [0.4073, 0.9103],
                 [0.9636, 0.5924]])
                
In  [5]: gumbel_sigmoid(x, hard=True)
Out [5]: tensor([[0., 0.],
                 [0., 1.],
                 [1., 1.]])

Releases

No releases published

Packages

No packages published

Languages