gumbel_softmax_kernel.cu 6.4 KB