SimThreadExponential

产品支持情况

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

功能说明

  • 算子功能:生成服从参数为lambda的指数分布随机数,并将其填充到self张量中。

  • 计算公式:

    f(x)=−1λln⁡(1−u),u∼Uniform(0,1]f(x) = -\frac{1}{\lambda} \ln(1 - u), u \sim \text{Uniform}(0, 1]

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
self 输入/输出 输入输出tensor。 FLOAT16、BFLOAT16、FLOAT32 ND
count 属性 生成的随机数元素个数。 INT64 ND
lambda 属性 公式中的lambda。 DOUBLE -
seed 属性 随机数种子。 INT64 -
offset 属性 随机数偏置值。 INT64 -

约束说明

  • lambd 必须大于 0
  • offset 必须为 4 的倍数(Ascend 950)
  • count 必须等于 self 张量的元素总数(Ascend 950)
  • 仅支持 ND 格式

调用说明

调用方式 样例代码 说明
aclnn接口 test_sim_thread_exponential 通过aclnnSimThreadExponential接口方式调用SimThreadExponential算子。