news 2026/7/16 16:26:37

CANN/asc-devkit SIMD C API LeakyReLU函数

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
CANN/asc-devkit SIMD C API LeakyReLU函数

asc_leakyrelu

【免费下载链接】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 训练系列产品:不支持

功能说明

头文件路径:"c_api/reg_compute/reg_vector.h"

按元素执行Leaky ReLU(Leaky Rectified Linear Unit)操作。Leaky ReLU带泄露性整流函数是一种人工神经网络中常用的激活函数,其数学表达式为:

$$ f(x_i) = \begin{cases} x_i & \text{if } x_i ≥ \text{ }0 \ \ ax_i & \text{if } x_i < \text{ }0 \ \end{cases} $$

函数原型

__simd_callee__ inline void asc_leakyrelu(vector_float& dst, vector_float src, float value, vector_bool mask) __simd_callee__ inline void asc_leakyrelu(vector_half& dst, vector_half src, half value, vector_bool mask)

参数说明

参数名输入/输出描述
dst输出目的操作数(矢量数据寄存器)。
src输入源操作数(矢量数据寄存器)。
value输入源操作数(标量)。
mask输入源操作数掩码(掩码寄存器),用于指示在计算过程中哪些元素参与计算。对应位置为1时参与计算,为0时不参与计算。mask未筛选的元素在输出中置零。

矢量数据寄存器和掩码寄存器的详细说明请参见reg数据类型定义。

返回值说明

流水类型

PIPE_V

约束说明

调用示例

vector_half dst; vector_half src; half value = 0.1; vector_bool mask = asc_create_mask_b16(PAT_ALL); asc_loadalign(src, src_addr); // src_addr是外部输入的UB内存空间地址。 asc_leakyrelu(dst, src, value, mask);

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/7/16 16:26:03

Linux系统下编译安装OpenCV与首个视觉程序实战

1. 为什么选择源码编译安装OpenCV在Linux系统下安装OpenCV通常有两种方式&#xff1a;直接使用包管理器安装预编译版本&#xff0c;或者从源码编译安装。很多新手可能会觉得第一种方式更简单&#xff0c;但实际开发中&#xff0c;源码编译安装才是更推荐的做法。我自己在多个视…

作者头像 李华
网站建设 2026/7/16 16:24:53

编译原理-5-LL(1)预测分析表的构造与实战

1. LL(1)预测分析表的核心概念第一次接触LL(1)预测分析表时&#xff0c;很多人会被各种集合计算绕晕。其实它的核心思想很简单&#xff1a;让语法分析器像查地图一样做决策。想象你开车来到一个岔路口&#xff0c;导航会告诉你"看到麦当劳标志就左转&#xff0c;看到加油站…

作者头像 李华
网站建设 2026/7/16 16:24:01

Boost升压电路布局与设计实战指南

1. Boost电路布局的本质解析Boost升压电路作为开关电源三大基础拓扑之一&#xff0c;其核心原理看似简单&#xff0c;但实际布局中90%的问题都源于对基本原理的理解偏差。我们先从能量流动的角度重新认识这个经典电路。当MOS管导通时&#xff0c;输入电源Vin通过电感L1形成回路…

作者头像 李华
网站建设 2026/7/16 16:24:01

Vision Transformers CIFAR-10实战:5天从零到精通的终极指南

Vision Transformers CIFAR-10实战&#xff1a;5天从零到精通的终极指南 【免费下载链接】vision-transformers-cifar10 Lets train vision transformers (ViT) for cifar 10 / cifar 100! 项目地址: https://gitcode.com/gh_mirrors/vi/vision-transformers-cifar10 你…

作者头像 李华
网站建设 2026/7/16 16:23:29

线性代数(5)—— 秩的几何直观与子空间维度

1. 秩的几何直观&#xff1a;从空间压缩到信息维度 第一次接触"秩"这个概念时&#xff0c;我盯着数学教材上的定义发呆了半小时——"矩阵中非零子式的最高阶数"&#xff1f;这抽象定义就像天书。直到导师用投影仪演示了一个三维物体在二维平面上的投影&…

作者头像 李华