ComplexAbs

产品支持情况

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

功能说明

• 算子功能：计算输入复数张量中每一个元素的绝对值（模长）。

• 计算公式：

  对于复数输入 $x_i=a_i+b_i\cdot j$，输出其模长：

$$y_i=|x_i|=\sqrt{a_i^2+b_i^2}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行复数绝对值计算的入参，公式中的输入张量x。	COMPLEX32、COMPLEX64	ND
y	输出	复数绝对值计算的出参，公式中的输出张量y。输出数据类型由输入数据类型决定：输入为COMPLEX64时输出FLOAT，输入为COMPLEX32时输出FLOAT16。	FLOAT、FLOAT16	ND
Tout	属性	表示输出的数据类型，兼容TensorFlow ComplexAbs算子保留，当前不使用，输出数据类型由输入数据类型推导得到。	Type，默认值为DT_FLOAT	-

约束说明

• 输入只支持COMPLEX32、COMPLEX64类型，不支持其他数据类型。
• 输出数据类型由输入数据类型决定，不支持显式指定：COMPLEX64对应输出FLOAT，COMPLEX32对应输出FLOAT16。
• 输入与输出的shape一致，支持动态Shape（DynamicShape）与动态维度（DynamicRank）。
• 数据格式支持ND，支持非连续的Tensor，维度不大于8。

调用说明

调用方式	调用样例	说明
图模式调用	算子IR	通过算子IR构图方式调用ComplexAbs算子。