昇腾社区首页
中文
注册

Caxpy

功能描述

  • asdBlasMakeCaxpyPlan

    接口功能:初始化该句柄对应的Caxpy算子配置。

  • asdBlasCaxpy

    接口功能:实现复数向量乘以一个复数标量后,再加上一个复数向量

计算公式

asdBlasCaxpy
  • 计算公式:
  • 示例:

    输入“x”为:

    [3.0 + 4.0j, 4.0 + 4.0j]

    输入“alpha”为:

    3.0 + 4.0j

    输入“y”为:

    [3.0 + 4.0j, 4.0 + 4.0j]

    调用“Caxpy”算子后,

    输出“y”为:

    [-4.0 + 28.0j, 0.0 + 32.0j]

接口原型

  • AspbStatus asdBlasMakeCaxpyPlan(asdBlasHandle handle)
  • AspbStatus asdBlasCaxpy(asdBlasHandle handle, const int64_t n, const std::complex<float> *alpha, aclTensor* x,int64_t incx, aclTensor* y, int64_t incy)

参数列表

  • asdBlasMakeCaxpyPlan

    参数名称

    Input/Output

    类型

    描述

    handle

    Input

    asdBlasHandle

    Caxpy算子的句柄。

  • asdBlasCaxpy

    参数名称

    Input/Output

    类型

    描述

    handle

    Input

    asdBlasHandle

    Caxpy算子的句柄。

    n

    Input

    const int64_t

    向量中复数元素的个数。

    alpha

    Input

    const std::complex<float> *

    公式中的alpha,输入的复数标量,数据类型支持COMPLEX64。

    x

    Input

    aclTensor*

    公式中的x,Device侧的Tensor,数据类型支持COMPLEX64,数据格式支持ND,shape为[n]。

    incx

    Input

    int64_t

    向量x相邻元素间的内存地址偏移量(当前约束为1)。

    y

    Input/Output

    aclTensor*

    公式中的y,Device侧的Tensor,数据类型支持COMPLEX64,数据格式支持ND,shape为[n]。

    incy

    Input

    int64_t

    向量y相邻元素间的内存地址偏移量(当前约束为1)。

规格约束

  • asdBlasMakeCaxpyPlan:无。
  • asdBlasCaxpy
    • 输入的元素个数n当前覆盖支持[1,6.71e+06];
    • 算子输入shape为[n]、[n],输出shape为[n];
    • 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。