文件最后提交记录最后更新时间
7 天前
2 个月前
1 个月前
8 天前
8 天前
1 个月前
3 个月前
4 个月前
8 天前
README

Complex

产品支持情况

产品 是否支持
Ascend 950PR&950DT系列产品 √
Atlas A3系列产品 √
Atlas A2系列产品 √
Atlas 200I/500 A2推理产品 ×
Atlas推理系列产品 √
Atlas训练系列产品 √

功能说明

  • 算子功能:根据实部real和虚部imag构建复数张量。

  • 计算公式:

    out[i]=real[i]+imag[i]∗j\text{out}[i] = \text{real}[i] + \text{imag}[i] * \mathrm{j}

    其中,j\mathrm{j}为虚数单位,real和imag按broadcast后的shape参与计算。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
real 输入 复数的实部,公式中的real。 FLOAT、FLOAT16、DOUBLE ND
imag 输入 复数的虚部,公式中的imag。 FLOAT、FLOAT16、DOUBLE ND
out 输出 复数张量,公式中的out。 COMPLEX64、COMPLEX32、COMPLEX128 ND

约束说明

  • real和imag的数据类型必须一致,维度均不能超过8维,shape需满足broadcast关系。
  • out的shape必须是real与imag broadcast之后的shape。
  • 在Atlas A3系列产品、Ascend 950PR&950DT系列产品上,当输入为FLOAT、FLOAT16、DOUBLE时,输出数据类型分别应为COMPLEX64、COMPLEX32、COMPLEX128。
  • real、imag和out均支持非连续Tensor。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_complex 通过aclnnComplex接口方式调用Complex算子。