Complex

产品支持情况

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

功能说明

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

• 计算公式：

  $$\text{out}[i]=\text{real}[i]+\text{imag}[i]\ast \mathrm{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算子。