tf.random.stateless_truncated_normal