SquareSumAll

产品支持情况

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

功能说明

• 算子功能：对两个形状相同的输入张量x1、x2分别求全元素平方和，输出两个标量y1、y2。

• 计算公式：

$$y_1=\sum _{i=0}^{N-1}{x}_{1,i}^2,y_2=\sum _{i=0}^{N-1}{x}_{2,i}^2$$

其中 $N$ 为输入张量的元素总数。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x1	输入	表示第一路待求平方和的张量，对应公式中的x1。	FLOAT	FRACTAL_Z、C1HWNCoC0、NC1HWC0、ND、NCHW、NHWC
x2	输入	表示第二路待求平方和的张量，对应公式中的x2。	FLOAT	FRACTAL_Z、C1HWNCoC0、NC1HWC0、ND、NCHW、NHWC
y1	输出	表示x1全部元素的平方和，对应公式中的y1。 输出为0维Tensor（rank为0），含1个FLOAT元素。	FLOAT	FRACTAL_Z、C1HWNCoC0、NC1HWC0、ND
y2	输出	表示x2全部元素的平方和，对应公式中的y2。 数据类型、shape与输出y1保持一致。	FLOAT	FRACTAL_Z、C1HWNCoC0、NC1HWC0、ND

产品差异说明

下表中的“输入格式→输出格式”同时适用于x1→y1和x2→y2，两路输入必须使用相同的数据格式。

产品	静态shape格式组合	动态shape格式组合	rank限制
Ascend 950PR&950DT系列产品	ND→ND、NCHW→ND、NHWC→ND	ND→ND、NCHW→ND、NHWC→ND	ND输入的rank取值范围为[0, 8]；NCHW、NHWC输入的rank必须为4。
Atlas A3系列产品 Atlas A2系列产品 Atlas 200I/500 A2推理产品 Atlas推理系列产品 Atlas训练系列产品	FRACTAL_Z→FRACTAL_Z、C1HWNCoC0→C1HWNCoC0、NC1HWC0→NC1HWC0、ND→ND	ND→ND	输入的rank取值范围为[0, 8]。

约束说明

• 图编译期用-1表示未知维度、[-2]表示未知rank；两个输出始终为rank-0标量。执行期须先解析输入shape，再满足下述非空、同shape及格式约束。

• x1与x2不支持空Tensor；rank为0时表示含1个元素的标量输入，rank大于0时每一维长度必须大于0。

• x1与x2的执行期rank及每一维长度必须完全相同，不支持广播。shape[2, 3]与[6]即使元素数相同也不合法。

• 算子节点的四个参数均须使用FLOAT类型，数据格式必须满足产品差异说明中的合法组合。

• NaN、Inf及有限值平方上溢按FLOAT语义自然传播；两路累加器与两个输出互不污染，x1中的非有限值不会影响y2，反之亦然。

• 本算子仅提供GE图模式注册，构建配置为aclnn_exclude，不提供aclnn接口。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_square_sum_all	通过算子IR构图方式调用SquareSumAll算子。