... your own. torch.nn.functional.gumbel_softmax(logits, tau=1, hard=False, eps=1e-10, dim=- 1) [source] Samples from the Gumbel-Softmax distribution ( Link ...
確定! 回上一頁