test_gumbel_softmax_op.py 7.9 KB