Asum

产品支持情况

功能说明

接口功能：

• asdBlasMakeAsumPlan：初始化该句柄对应的Asum算子配置。
• asdBlasSasum：对输入的所有元素取绝对值后求和。
• asdBlasScasum：对输入的所有元素取绝对值后求和。输入元素为复数。

计算公式

• asdBlasSasum的计算公式：
  • 示例：

    输入“x”为：

    [1, 2, -3, 4]

    调用“Sasum”算子后，

    输出“result”为：

    10

• asdBlasScasum的计算公式：
  • 示例：

    输入“x”为：

    [1+2i, 2-2i, -3+3i, 4-3i]

    调用“Scasum”算子后，

    输出“result”为：

    20

函数原型

• AspbStatus asdBlasMakeAsumPlan(asdBlasHandle handle)
• AspbStatus asdBlasSasum(asdBlasHandle handle, const int64_t n, aclTensor *x, const int64_t incx, aclTensor *result)
• AspbStatus asdBlasScasum(asdBlasHandle handle, const int64_t n, aclTensor *x, const int64_t incx, aclTensor *result)

参数说明

• asdBlasMakeAsumPlan

  参数名称	Input/Output	类型	描述
  handle	Input	asdBlasHandle	Asum算子的句柄。

• asdBlasSasum & asdBlasScasum

  参数名称	Input/Output	类型	描述
  handle	Input	asdBlasHandle	asum算子的句柄。
  n	Input	const int64_t	总的元素个数。
  x	Input	aclTensor *	公式中的x，Device侧的Tensor，数据格式支持ND，shape为[n]。 Sasum下数据类型支持FLOAT32； Scasum下数据类型支持COMPLEX64。
  incx	Input	const int64_t	相邻元素间的内存地址偏移量（当前约束为1）。
  result	Output	aclTensor *	公式中的result，Device侧的Tensor，数据类型支持FLOAT32，只包含一个元素，数据格式支持ND，shape为[1]。

返回值说明

返回值相关说明参见返回值。

约束说明

• asdBlasMakeAsumPlan：无。
• asdBlasSasum
  • 输入的元素个数n当前覆盖支持[1，6.71e+06]；
  • 算子输入shape为[n]，输出shape为[1]；
  • 算子实际计算时，不支持ND高维度运算（不支持维度≥3的运算）。
• asdBlasScasum
  • 输入的元素个数n当前覆盖支持[1，6.71e+06]；
  • 算子输入shape为[n]，输出shape为[1]；
  • 算子实际计算时，不支持ND高维度运算（不支持维度≥3的运算）。

调用示例

算子的调用示例参见Asum。