Gumbel softmax being removed?

whoab · May 27, 2019, 2:56pm

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?

zyser · February 15, 2020, 8:12am

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

mruberry · February 15, 2020, 8:20am

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.

zyser · February 16, 2020, 8:40am

Thanks, I will use it then.