文件最后提交记录最后更新时间
19 天前
15 天前
5 个月前
15 天前
15 天前
15 天前
5 个月前
19 天前
README

ForeachNorm

产品支持情况

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

功能说明

  • 算子功能:对输入张量列表的每个张量进行范数运算。
  • 计算公式:
    • Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:

      x=[x0,x1,...xn−1]x = [{x_0}, {x_1}, ... {x_{n-1}}]\\

      y=(∑i=0n−1∣xi∣p)1p(i=0,1,...n−1)y = \left(\sum_{i=0}^{n-1}|x_i|^{p}\right)^{\frac{1}{{p}}} (i=0,1,...n-1)

    • Ascend 950PR/Ascend 950DT:

      x=[x0,x1,...xn−1]x = [{x_0}, {x_1}, ... {x_{n-1}}]\\

      • scalar = +Inf:

        y=max⁡∣xi∣y = \max|x_i|

      • int(scalar) = 1:

        y=∑i=0n−1∣xi∣y = \sum_{i=0}^{n-1}|x_i|

      • other:

        y=(∑i=0n−1∣xi∣2)12y = \left(\sum_{i=0}^{n-1}|x_i|^{2}\right)^{\frac{1}{{2}}}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 表示进行范数运算的输入张量列表,对应公式中的`x`。该参数中所有Tensor的数据类型保持一致。 FLOAT32、FLOAT16、BFLOAT16 ND
scalar 输入 表示进行范数运算的阶数,对应公式中的`p`。 FLOAT32、INT64 ND
y 输出 表示进行范数运算的输出张量列表,对应公式中的`y`。数据类型、数据格式与入参`x`的数据类型、数据格式一致,shape size为1。 FLOAT32、FLOAT16、BFLOAT16 ND
  • Ascend 950PR/Ascend 950DT:

    • 输入x、输出y支持的最大长度为256个。
    • 输入scalar的取值,仅支持1或者2。
  • Kirin X90/Kirin 9030处理器系列产品:x和y的数据类型不支持BFLOAT16。

约束说明

  • 输出不支持非连续Tensor。

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_foreach_norm 通过aclnnForeachNorm接口方式调用ForeachNorm算子。
图模式 - 通过算子IR构图方式调用ForeachNorm算子。