ReduceSumOp

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
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 A3系列产品、Ascend 950PR&950DT系列产品：数据类型支持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 A3系列产品、Ascend 950PR&950DT系列产品：数据类型支持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 A3系列产品、Ascend 950PR&950DT系列产品：数据类型支持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算子。