StatelessUniform

产品支持情况

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

功能说明

• 算子功能：基于Philox4x32-10伪随机数生成算法，生成服从均匀分布的随机数张量，随机值范围为[from, to)。

• 计算公式：

  $$y_i=u_i\times (to-from)+from,u_i\sim \text{Uniform}(0,1]$$

  其中 $u_i$ 由Philox4x32-10算法生成，归一化方式与竞品curand_uniform一致：$u=x\times 2^{-32}+2^{-33}$。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
shape	输入	输出张量的形状，1-D tensor。	INT64	ND
seed	输入	Philox算法的随机数种子，0-D标量。	INT64	ND
offset	输入	Philox算法的偏移量，0-D标量。必须是4的倍数。	INT64	ND
from	输入	均匀分布随机范围的下界（包含），0-D标量。	DOUBLE	ND
to	输入	均匀分布随机范围的上界（不包含），0-D标量。	DOUBLE	ND
y	输出	输出指定形状的均匀分布随机值，值域为[from, to)。	FLOAT16、BFLOAT16、FLOAT32	ND
dtype	属性	输出数据类型，默认为FLOAT32。	Type	-

约束说明

• offset必须是4的倍数。
• 输出张量维度支持0~8维。
• from必须小于等于to，且to - from不能超出输出数据类型的表示范围。

调用说明

调用方式	样例代码	说明
aclnn调用	test_aclnn_stateless_uniform	通过aclnnInplaceUniform接口方式调用StatelessUniform算子。
图模式调用	test_geir_stateless_uniform	通过算子IR构图方式调用StatelessUniform算子。