大概可以參考pytorch裡面torch.nn.functional.gumbel_softmax的實現:def gumbel_softmax(logits, tau=1, hard=False, eps=1e-10, dim=-1): if eps != 1e-...
確定! 回上一頁