StatelessRandom

产品支持情况

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

功能说明

  • 算子功能:返回从[from, to - 1]范围中抽取离散均匀分布的随机数。
  • 计算公式:

    output[i]=X%(to−from)+fromoutput[i] = X \% (to - from) + from

    其中X为Philox随机数生成器产生的uint32随机数。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
shape 输入 输出张量的形状维度。 INT64 ND
seed 输入 随机数种子。 INT64 ND
offset 输入 随机数偏移量,必须是4的倍数。 INT64 ND
from 输入 可选参数,随机数范围的下界。 INT64 ND
to 输入 可选参数,随机数范围的上界。 INT64 ND
y 输出 输出张量,包含指定范围内的随机值。 FLOAT、FLOAT16、BFLOAT16、INT64、INT32、INT16、INT8、UINT8、BOOL ND
dtype 属性 输出张量的数据类型。 DataType -

约束说明

  • offset是4的倍数。
  • 输出张量维度支持0~8维
  • from必须小于to, from/to指定范围不可超出self数据类型的范围。

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_inplace_random 通过aclnnInplaceRandomaclnnInplaceRandomTensor接口方式调用StatelessRandom算子。
图模式调用 test_geir_stateless_random 通过算子IR构图方式调用StatelessRandom算子。