昇腾社区首页
中文
注册

Csrot

功能描述

  • asdBlasMakeRotPlan

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

  • asdBlasCsrot

    接口功能:对输入复向量组(x,y)进行旋转

计算公式

asdBlasCsrot
  • 计算公式:

    其中c为旋转角度余弦值,s为旋转角度正弦值,x和y是复数向量。

  • 示例:

    输入“x”为:

    [3.0 + 4.0j, 2.0 + 1.0j]

    输入“y”为:

    [1.0 + 1.0j, 3.0 + 3.0j]

    输入“c”为:

    输入“s”为:0.5

    调用“Csrot”算子后,

    输出“x”为:

    [3.098076+3.9641016j, 3.232051+2.3660254j]

    输出“y”为:

    [-0.6339746-1.1339746j, 1.5980761+2.098076j]

接口原型

  • AspbStatus asdBlasMakeRotPlan(asdBlasHandle handle)
  • AspbStatus asdBlasCsrot(asdBlasHandle handle, const int64_t n, aclTensor *x, const int64_t incx, aclTensor *y,const int64_t incy, const float *c, const float *s)

参数列表

  • asdBlasMakeRotPlan

    参数名称

    Input/Output

    类型

    描述

    handle

    Input

    asdBlasHandle

    csrot算子的句柄。

  • asdBlasCsrot

    参数名称

    Input/Output

    类型

    描述

    handle

    Input

    asdBlasHandle

    csrot算子的句柄。

    n

    Input

    const int64_t

    向量中复数元素的个数。

    x

    Input/Output

    aclTensor *

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

    incx

    Input

    const int64_t

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

    y

    Input/Output

    aclTensor *

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

    incy

    Input

    const int64_t

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

    c

    Input

    const float *

    旋转矩阵的余弦值指针。

    s

    Input

    const float *

    旋转矩阵的正弦值指针。

规格约束

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