Colwise_mul

产品支持情况

功能描述

接口功能

• asdBlasMakeColwiseMulPlan：初始化该句柄对应的ColwiseMul算子配置。
• asdBlasColwiseMul：复数矩阵与复数向量按列逐点乘，返回一个和输入矩阵同样形状大小的复数矩阵。

计算公式

• asdBlasColwiseMul的计算公式：

  

• 示例：

  输入“A”为：

  [ [ 1+1i, 1+1i ],
    [ 2+2i, 2+2i ] ]

  输入“X”为：

  [ 1+1i, 2+2i ]

  调用“asdBlasColwiseMul”算子后，

  输出“result”为：

  [ [ 0+2i, 0+2i ],
    [ 0+8i, 0+8i ] ]

函数原型

• AspbStatus asdBlasMakeColwiseMulPlan(asdBlasHandle handle)
• AspbStatus asdBlasColwiseMul(asdBlasHandle handle, const int64_t m, const int64_t n, aclTensor *mat, aclTensor *vec,aclTensor *result)

参数说明

• asdBlasMakeColwiseMulPlan

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

• asdBlasColwiseMul

  参数名称	Input/Output	类型	描述
  handle	Input	asdBlasHandle	ColwiseMul算子的句柄。
  m	Input	const int64_t	矩阵mat的行数，向量vec的元素个数。
  n	Input	const int64_t	矩阵mat的列数。
  mat	Input	aclTensor *	公式中的A，Device侧的tensor，数据类型支持COMPLEX64 ，数据格式支持ND，shape为[m，n]。
  vec	Input	aclTensor *	公式中的x，复数向量，Device侧的tensor，数据类型支持COMPLEX64 ，数据格式支持ND，shape为[m]。
  result	Output	aclTensor *	公式中的result，Device侧的tensor，数据类型支持COMPLEX64，数据格式支持ND，shape为[m，n]，数据格式、类型、shape与入参mat一致。

返回值说明

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

约束说明

• asdBlasMakeColwiseMulPlan：无。
• asdBlasColwiseMul
  • ；
  • 算子输入shape为[m，n]、[m]，输出shape为[m，n]；
  • 算子实际计算时，不支持ND高维度运算（不支持维度≥3的运算）。

调用示例

算子的调用示例参见Colwise_mul。