test_gumbel_softmax_op.py 9.3 KB