PRelu

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：一种激活函数，当Tensor中value大于0，取该value，小于0时取权重与value的乘积。

• 计算公式：

$$ou{t}_i=\mathrm{P}\mathrm{R}\mathrm{e}\mathrm{l}\mathrm{u}(x_i)=\max (0,x_i)+weight\times \min (0,x_i)$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行PRelu计算的入参，公式中的x。	FLOAT、FLOAT16、BFLOAT16	ND
weight	输入	计算的权重参数，元素个数为1或者元素个数与x输入的channel数量一致，channel为输入维度的第二维度。	FLOAT、FLOAT16、BFLOAT16	ND
out	输出	PRelu计算的出参，对应计算公式中的out。	FLOAT、FLOAT16、BFLOAT16	ND

• Atlas 推理系列产品、Atlas 训练系列产品：数据类型支持FLOAT、FLOAT16。

约束说明

无

调用说明

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