tf.random.stateless_parameterized_truncated_normal