CANN / ops-math ReduceSum算子

ReduceSumOp

【免费下载链接】ops-math 本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。 项目地址: https://gitcode.com/cann/ops-math

产品支持情况

功能说明

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

参数说明

• 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。

约束说明

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

调用示例

【免费下载链接】ops-math 本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。 项目地址: https://gitcode.com/cann/ops-math