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