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

样例:

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