CANN/asc-devkit Swish激活函数API

Swish【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品支持Atlas A2 训练系列产品/Atlas A2 推理系列产品支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品AI Core支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明在神经网络中Swish是一个重要的激活函数。计算公式如下其中β为常数函数原型template typename T, bool isReuseSource false __aicore__ inline void Swish(const LocalTensorT dstLocal, const LocalTensorT srcLocal, uint32_t dataSize, const T scalarValue)参数说明表1模板参数说明参数名描述T操作数的数据类型。支持的数据类型为half、float。isReuseSource是否允许修改源操作数。该参数预留传入默认值false即可。表2接口参数说明参数名输入/输出描述dstLocal输出目的操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。srcLocal输入源操作数。源操作数的数据类型需要与目的操作数保持一致。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。dataSize输入实际计算数据元素个数。scalarValue输入激活函数中的β参数。支持的数据类型为half、float。β参数的数据类型需要与源操作数和目的操作数保持一致。返回值说明无约束说明操作数地址偏移对齐要求请参见通用说明和约束。不支持源操作数与目的操作数地址重叠。当前仅支持ND格式的输入不支持其他格式。调用示例// dstLocal: 存放Swish计算结果的Tensor // srcLocal存放Swish计算的输入Tensor // scalarValue: 激活函数中的β参数 // 参与计算的元素个数为32β参数为-1.702 AscendC::Swish(dstLocal, srcLocal, 32, -1.702);结果示例如下输入数据(srcLocal): [-4. -3.7419355 -3.483871 -3.2258065 -2.967742 -2.7096775 -2.451613 -2.1935484 -1.9354838 -1.6774193 -1.4193548 -1.1612903 -0.9032258 -0.6451613 -0.38709676 -0.12903225 0.12903225 0.38709676 0.6451613 0.9032258 1.1612903 1.4193548 1.6774193 1.9354838 2.1935484 2.451613 2.7096775 2.967742 3.2258065 3.483871 3.7419355 4. ] 输出数据(dstLocal): [-0.00441472 -0.00640367 -0.00924141 -0.01325663 -0.01887952 -0.02665107 -0.03720944 -0.05122496 -0.06923404 -0.09129371 -0.11635891 -0.14131825 -0.15980731 -0.16135658 -0.13200021 -0.05746017 0.07157208 0.25509655 0.48380467 0.74341846 1.019972 1.3029958 1.5861256 1.8662498 2.1423235 2.4144034 2.6830263 2.9488626 3.2125497 3.4746296 3.735532 3.995585 ]【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考