CANN/ops-math掩码缩放算子

MaskedScale

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math

产品支持情况

功能说明

• 算子功能：完成elementwise计算

• 计算公式：

  $$ out = self \times mask \times scale $$

调用说明

当前算子作为自定义算子通过图模式下发执行，aclnnMaskedScale两段式接口仅供内部使用，不对外暴露。

算子输入输出

• 参数说明：

  • self(计算输入)：公式中的输入self，Device侧Tensor。

    • Ascend 950PR/Ascend 950DT ：数据类型支持FLOAT16、BFLOAT16、FLOAT，数据格式支持ND。
    • Atlas A3 训练系列产品/Atlas A3 推理系列产品 ：数据类型支持FLOAT16、BFLOAT16、FLOAT，数据格式支持ND。

  • mask(计算输入)：公式中的mask，Device侧Tensor，shape需要与self一致。

    • Ascend 950PR/Ascend 950DT ：数据类型支持UINT8、INT8、FLOAT16、FLOAT，数据格式支持ND。
    • Atlas A3 训练系列产品/Atlas A3 推理系列产品 ：数据类型支持UINT8、INT8、FLOAT16、FLOAT，数据格式支持ND。

  • scale(float, 计算输入)：标量缩放系数，数据类型支持FLOAT（非Tensor）。

  • y(计算输出)：公式中的out，Device侧Tensor，数据类型和shape需要与self一致。

    • Ascend 950PR/Ascend 950DT ：数据类型支持FLOAT16、BFLOAT16、FLOAT，数据格式支持ND。
    • Atlas A3 训练系列产品/Atlas A3 推理系列产品 ：数据类型支持FLOAT16、BFLOAT16、FLOAT，数据格式支持ND。

约束说明

• 确定性计算：
  • MaskedScale默认确定性实现。
• shape约束：
  • self、mask和y的shape需要一致。
  • 当前实现的元素总数不能超过uint32_t可表示范围。
  • 输入/属性命名采用self、mask、scale、y，与内部 aclnn 接口保持一致。
  • FLOAT16 路径中scale按 half 标量参与向量乘，存在标量精度截断；FLOAT/BFLOAT16 路径按 float 标量计算。

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math