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