Skip to content

sirluk/gumbel_sigmoid

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

gumbel_sigmoid

A pytorch implementation of the gumbel sigmoid gate (paper)

example

logits = torch.randn(1000)
out = gumbel_sigmoid(logits, temp=0.6)

gumbel softmax output density for normally distributed inputs (temperature = 0.6)

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages