StatelessTruncatedNormalV2

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	×
Atlas A2 训练系列产品/Atlas A2 推理系列产品	×
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

• 算子功能：无状态版截断正态分布随机数生成器。使用Philox4x32算法，以key和counter为输入生成均匀分布的uint32随机数，通过Box-Muller变换将均匀分布转换为标准正态分布，生成服从标准正态分布N(0,1)的随机数，绝对值大于2.0的样本被截断丢弃并重新采样。
• 无状态特性：随机数生成完全由外部传入的key/counter决定，相同输入保证产生相同输出（确定性）。
• 对标：TensorFlow tf.raw_ops.StatelessTruncatedNormalV2

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
shape	输入	输出张量的形状	INT32、INT64	ND
key	输入	Philox随机数生成器密钥，shape=[1]	UINT64	ND
counter	输入	Philox随机数生成器计数器，shape=[2]	UINT64	ND
alg	输入	随机数算法ID（1=Philox，当前仅支持Philox）	INT32	ND
dtype	属性	指定输出的数据类型（0=float32, 1=float16, 27=bfloat16 ,11=float64）	INT	ND
y	输出	生成的截断正态分布随机数序列	FLOAT16、FLOAT、BF16、FLOAT64	ND

约束说明

• alg参数当前仅支持Philox算法（alg=1）
• key的shape必须为[1]
• counter的shape必须为[2]（Philox算法要求）
• 输出shape中元素数为0时直接返回空tensor

调用说明

调用方式	样例代码	说明
图模式调用	test_geir_stateless_truncated_normal_v2	通过算子IR构图方式调用StatelessTruncatedNormalV2算子。