tf.random.stateless_gamma