昇腾社区首页
中文
注册

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的运算)。