ReduceProd

产品支持情况

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

功能说明

算子功能:返回输入tensor中所有元素的乘积。

参数说明

  • self(aclTensor*, 计算输入):Device侧的aclTensor,支持非连续的Tensor数据格式支持ND。

    • Atlas 推理系列产品、Atlas 训练系列产品:数据类型支持FLOAT16、FLOAT、DOUBLE、INT8、UINT8、INT16、INT32、INT64、BOOL、COMPLEX64、COMPLEX128。输入为空tensor时,输出类型不能是复数类型COMPLEX64和COMPLEX128。
    • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件、Atlas A3 训练系列产品/Atlas A3 推理系列产品、昇腾950 AI处理器:数据类型支持FLOAT16、FLOAT、DOUBLE、INT8、UINT8、INT16、INT32、INT64、BOOL、COMPLEX64、COMPLEX128、BFLOAT16。输入为空tensor时,输出类型不能是复数类型COMPLEX64和COMPLEX128。
  • dtype(const aclDataType, 计算输入):Host侧的aclDataType,输出tensor的数据类型,需要与out的数据类型一致。

  • out(aclTensor*, 计算输出):Device侧的aclTensor,数据类型与dtype一致,若dtype未传入则数据类型与self一致。支持非连续的Tensor数据格式支持ND。输出shape为(1)。

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

约束说明

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

调用示例

调用方式 样例代码 说明
aclnn接口 test_aclnn_prod 通过aclnnProd接口方式调用ReduceProd算子。
aclnn接口 test_aclnn_prod_dim 通过aclnnProdDim接口方式调用ReduceProd算子。