aclnnSquaredRelu

产品支持情况

[object Object]undefined

功能说明

• 接口功能：

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

• 计算公式：

  $$y_i=(ReLU(x_i))^2$$

  其中ReLU的计算公式：

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

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

函数原型

每个算子分为，必须先调用“aclnnSquaredReluGetWorkspaceSize”接口获取入参并根据计算流程计算所需workspace大小，再调用“aclnnSquaredRelu”接口执行计算。

aclnnSquaredReluGetWorkspaceSize

• 参数说明：

  [object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。 第一段接口会完成入参校验，出现以下场景时报错：

  [object Object]

aclnnSquaredRelu

• 参数说明：

  [object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。

约束说明

• 确定性计算：

  • aclnnSquaredRelu默认确定性实现。

• 输入值为NaN，输出也为NaN，输入是Inf，输出也是Inf。

• 输入是-Inf，输出是0。

• 输入shape只支持维度1至8维，不在范围内执行报错。

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考。