test_gumbel_softmax_op.py 8.4 KB