test_gumbel_softmax_op.py 8.1 KB