StatelessRandomNormalV3

产品支持情况

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

功能说明

  • 算子功能:根据指定的均值(mean)和标准差(stdev)生成服从正态分布的随机数张量,使用基于计数器的Philox随机数生成算法,支持mean和stdev为标量或张量。

  • 计算公式:

    result=StatelessRandomNormalV2+Mul(std)+Add(mean)result = StatelessRandomNormalV2 + Mul(std) + Add(mean)

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
shape 输入 输出张量的形状。 INT64、INT32 ND
key 输入 用于基于计数器的随机数生成算法的秘钥。 UINT64 ND
counter 输入 用于基于计数器的随机数生成算法的初始计数值。 UINT64 ND
mean 输入 正态分布的均值,支持标量或与输出同形状的张量。 FLOAT ND
stdev 输入 正态分布的标准差,支持标量或与输出同形状的张量。 FLOAT ND
y 输出 输出指定形状的正态分布随机值。 FLOAT、BF16、FLOAT16 ND
dtype 属性 可选属性,指定输出数据类型。 INT -

约束说明

调用说明

调用方式 样例代码 说明
aclnn调用 test_aclnn_normal_float_float 通过aclnnNormalFloatFloat接口方式调用stateless_random_normal_v3算子。
aclnn调用 test_aclnn_normal_float_tensor 通过aclnnNormalFloatTensor接口方式调用stateless_random_normal_v3算子。
aclnn调用 test_aclnn_normal_tensor_float 通过aclnnNormalFloatTensor接口方式调用stateless_random_normal_v3算子。
aclnn调用 test_aclnn_normal_tensor_tensor 通过aclnnNormalTensorTensor接口方式调用stateless_random_normal_v3算子。