ReduceSumOp

产品支持情况

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

功能说明

算子功能：返回给定维度中输入张量每行的和。

参数说明

• self（aclTensor*, 计算输入）：Device侧的aclTensor，shape支持0-8维，支持非连续的Tensor，数据格式支持ND。

  • Atlas 推理系列产品、Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT32、INT8、INT16、INT32、INT64、UINT8、BOOL、DOUBLE、COMPLEX64、COMPLEX128。输入为空tensor时，输出类型不能是复数类型COMPLEX64和COMPLEX128。
  • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件、Atlas A3 训练系列产品/Atlas A3 推理系列产品、昇腾950 AI处理器：数据类型支持FLOAT16、FLOAT32、INT8、INT16、INT32、INT64、UINT8、BOOL、DOUBLE、COMPLEX64、COMPLEX128、BFLOAT16。输入为空tensor时，输出类型不能是复数类型COMPLEX64和COMPLEX128。

• dims（aclIntArray*, 计算输入）：Host侧的aclIntArray，指定reduce维度，数据类型支持INT64，取值范围为[-self.dim(), self.dim()-1]。

• keepDims（bool, 计算输入）：Host侧的BOOL值，指定是否在输出张量中保留输入张量的维度。

• dtype（aclDataType, 计算输入）：Device侧的aclDataType，指定输出张量的数据类型。

  • Atlas 推理系列产品、Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT32、INT8、 INT16、 INT32、 INT64、UINT8、BOOL、DOUBLE、COMPLEX64、COMPLEX128。
  • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件、Atlas A3 训练系列产品/Atlas A3 推理系列产品、昇腾950 AI处理器：数据类型支持FLOAT16、FLOAT32、INT8、 INT16、 INT32、 INT64、UINT8、BOOL、DOUBLE、COMPLEX64、COMPLEX128、BFLOAT16。

• out（aclTensor*, 计算输出）：Device侧的aclTensor，支持非连续的Tensor，数据格式支持ND。

  • Atlas 推理系列产品、Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT32、INT8、INT16、INT32、INT64、UINT8、BOOL、DOUBLE、COMPLEX64、COMPLEX128。
  • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件、Atlas A3 训练系列产品/Atlas A3 推理系列产品、昇腾950 AI处理器：数据类型支持FLOAT16、FLOAT32、INT8、INT16、INT32、INT64、UINT8、BOOL、DOUBLE、COMPLEX64、COMPLEX128、BFLOAT16。

约束说明

• 确定性计算：
  • 默认确定性实现。

调用示例

调用方式	样例代码	说明
aclnn接口	test_aclnn_reduce_sum	通过aclnnReduceSum接口方式调用ReduceSum算子。
图模式调用	test_geir_reduce_sum.cpp	通过GE IR图模式调用ReduceSum算子。