Simple replacement of argmax in Gumbel max trick with softmax function.
syyunn / categorical-reparameterization-with-gumbel-softmax Goto Github PK
View Code? Open in Web Editor NEWVisual proof of Gumbel-Softmax distribution approximating categorical distribution