非引流，实在觉得这两篇Gumbel Softmax讲的不错，也就没有自己写的必要了(并不是因为懒https://zhuanlan.zhihu.com/p/455084077https://zhu...