Gumbel-softmax其实可以看成gumbel-max + soft-argmax，其解决的问题是如何从一个离散分布中抽样，并且...
tau) y_soft = gumbels.softmax(dim) if hard: # Straight through. index =...