mindspore.ops.ComplexAbs

class mindspore.ops.ComplexAbs[源代码]

返回输入复数的模。

输入复数的形式为 \(a + bj\) ，其中 \(a\) 为实部， \(b\) 为虚部。

\[y = \sqrt{a^2+b^2}\]

警告

这是一个实验性API，后续可能修改或删除。

输入：

  • x (Tensor) - 复数Tensor，格式须为complex64或complex128。

输出：

Tensor。如果 x 的类型是complex64，则输出的类型是float32；如果 x 的类型是complex128，则输出的类型是float64。

异常：

  • TypeError - 输入 x 不是Tensor。

  • TypeError - 输入 x 不是complex64或complex128格式。

支持平台：

Ascend GPU CPU

样例：

>>> x = Tensor(np.asarray(np.complex(3+4j)), mindspore.complex64)
>>> complex_abs = ops.ComplexAbs()
>>> output = complex_abs(x)
>>> print(output)
5.0