SquaredRelu

产品支持情况

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

功能说明

• 接口功能：

  SquaredRelu函数是一个基于标准Relu函数的变体，其主要特点是对Relu函数的输出进行平方，常作为模型的激活函数。

• 计算公式：

  $$y_i=(Relu(x_i){)}^2$$

  其中Relu的计算公式：

  $$Relu(x_i)=max(0,x_i)$$

  其中，x是输入Tensor，y是输出Tensor，i是元素在Tensor中的位置索引。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	公式中的输入x。	BFLOAT16、FLOAT16、FLOAT	ND
y	输出	公式中的y。	BFLOAT16、FLOAT16、FLOAT	ND

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

约束说明

• 输入值为NaN，输出也为NaN，输入是Inf，输出也是Inf。
• 输入是-Inf，输出是0。
• 输入shape只支持维度1至8维，不在范围内执行报错。

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_squared_relu	通过aclnnSquaredRelu接口方式调用SquaredRelu算子。
图模式调用	-	通过算子IR构图方式调用SquaredRelu算子。