tau) y_soft = gumbels.softmax(dim) if hard: # Straight through. index =...
使用gumbel-max（注意是max不是softmax）能够等价于对softmax进行采样，但是还有一个问题就是argmax不...