torch.nn.functional.gumbel\_softmax =================================== .. currentmodule:: torch.nn.functional .. autofunction:: gumbel_softmax