比较与选择算子样例介绍
概述
本样例集介绍了比较与选择操作算子不同特性的典型用法,给出了对应的端到端实现。
算子开发样例
| 目录名称 | 功能描述 |
|---|---|
| compare | 本样例基于Compare逐元素比较两个tensor大小,如果比较后的结果为真,则输出结果的对应比特位为1,否则为0 |
| compare_result_stored_in_a_register | 本样例基于Compare(结果存入寄存器)接口进行比较操作,可用于逐元素比较两个tensor大小(比较后的结果为真,则输出结果的对应比特位为1,否则为0),并将计算结果存入寄存器中 |
| compares | 本样例基于Compares接口进行比较操作,可用于一个tensor逐元素与一个标量比较大小(比较后的结果为真,则输出结果的对应比特位为1,否则为0) |
| compares_flexible_scalar_argument_position | 本样例利用Ascend 950PR/Ascend 950DT产品的新增特性,实现了具有灵活标量位置能力的ComparesFlexibleScalarArgumentPosition算子,该算子实现与Compares相同,特殊在于该接口还支持标量参数位置在前和在后两种场景,其中标量输入还支持配置LocalTensor单点元素 |
| gather_mask_built_in_fixed_mode | 本样例基于GatherMask实现以内置固定模式对应的二进制为gather mask(数据收集的掩码),从源操作数中选取元素写入目的操作数中 |
| gather_mask_custom_mode | 本样例基于GatherMask基础API的用户自定义模式接口实现数据聚合,可用于以用户输入的Tensor数值对应的二进制为gather mask(数据收集的掩码),从源操作数中选取元素写入目的操作数中 |
| select | 本样例基于Select完成选择操作,可用于给定两个源操作数src0和src1,根据selMask(用于选择的Mask掩码)的比特位值选取元素,得到目的操作数dst |
| select_flexible_scalar_argument_position | 本样例基于Select实现对于给定的两个源操作数src0和scalar标量,根据selMask(用于选择的Mask掩码)的比特位值选取元素,得到目的操作数dst。选择的规则为:当selMask的比特位是1时,从src0中选取,比特位是0时选取scalar标量 |