企业官网建设流程全解析

FatreluMul

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

产品支持情况

功能说明

• 接口功能：

  将输入Tensor按照最后一个维度分为左右两个Tensor：x1和x2，对左边的x1进行Threshold计算，将计算结果与x2相乘。

• 计算公式：

  给定输入张量input，最后一维的长度为2d，进行以下计算：

  1. 将input分割为两部分：

     $$ x_1 = \text{input}[..., :d], \quad x_2 = \text{input}[..., d:] $$

  2. 对x1应用Threshold激活函数，定义如下：

     $$ \text{Threshold}(x, \text{threshold}) = \begin{cases} 0 & \text{if } x < \text{threshold} \ x & \text{if } x \geq \text{threshold} \end{cases} $$

     因此，计算：

     $$ x_1 = \text{Threshold}(x_1, \text{threshold}) $$

  3. 最终输出是x1和x2的逐元素乘积：

     $$ \text{out} = x_1 \times x_2 $$

参数说明

• Kirin X90/Kirin 9030处理器系列产品：不支持BFLOAT16。

约束说明

典型场景尾轴为16的倍数，当尾轴为非32Byte对齐时，建议走小算子拼接逻辑。

调用说明

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