文件最后提交记录最后更新时间
6 个月前
7 个月前
7 个月前
7 个月前
7 个月前
7 个月前
7 个月前
README

StatelessRandomChoiceWithMask

产品支持情况

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

功能说明

  • 算子功能: 根据输入x中值为true,选择出对应索引列表,其次根据seed, offset调用philox_random生成索引列表长度个随机数,利用随机数对索引列表做洗牌算法,得到随机交换后得索引列表,返回count个。 如果count为0,输出实际索引列表个。如果列表不足count个,填充0,补充至count个。输出mask标记该索引是否为有效索引。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入x值。 BOOL ND
count 输入 期望输出索引个数。 INT32 ND
seed 输入 获取随机种子。 INT64 ND
offset 输入 获取值的步长。 INT64 ND
out 输出 洗牌后的输入x值为true的索引。 INT32 ND
mask 输出 标记该位置的索引是否为有效索引。 BOOL ND

约束说明

无

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_stateless_random_choice_with_mask 通过算子IR构图方式调用StatelessRandomChoiceWithMask算子。