Gumbel softmax being removed?

Why this?

This function is here for legacy reasons, may be removed from nn.Functional in the future.

What is the newest place to get it versus nn functional?

Same question here, why there is a note Gumbel Softmax may be removed?

The PR that added the warning (https://github.com/pytorch/pytorch/pull/13339) has a little discussion about its origin. As best I can tell, that PR’s author thought the function might be implemented via a different package in the future. I don’t think you have to worry about gumbel_softmax going away.

1 Like

Thanks, I will use it then.