文件最后提交记录最后更新时间
13 天前
30 天前
30 天前
30 天前
12 天前
1 个月前
1 个月前
README

Prelu

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

  • 算子功能:计算输入张量的 PReLU 值。当输入元素大于 0 时输出该元素本身;当输入元素小于等于 0 时输出该元素与 weight 的乘积。

  • 计算公式:

yi={xi,xi>0xi×weight,xi≤0{y}_i = \begin{cases} x_i, & x_i > 0 \\ x_i \times weight, & x_i \le 0 \end{cases}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 待进行 Prelu 计算的输入张量。 FLOAT、FLOAT16、BFLOAT16 ND
weight 输入 Prelu 负半轴权重,元素个数为 1 或与输入通道数一致。 FLOAT、FLOAT16、BFLOAT16 ND
y 输出 Prelu 计算后的输出张量,shape 与 x 一致。 FLOAT、FLOAT16、BFLOAT16 ND

约束说明

  • xweighty 的数据类型需要一致。
  • y 的 shape 必须与 x 完全一致。
  • weight 的元素个数必须为 1 或与 x 的通道数一致。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_prelu.cpp 通过 aclnnPrelu 接口方式调用 Prelu 算子。

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
Han Chuang 个人开发者 Prelu 2026/06/05 Prelu 算子适配开源仓