mindspore.ops.Complex
- class mindspore.ops.Complex[源代码]
给定复数的实部与虚部,返回一个复数的Tensor。
警告
这是一个实验性API,后续可能修改或删除。
- 输入:
real (Tensor) - 实部的取值。数据类型:float32,float64。
imag (Tensor) - 虚部的取值。数据类型:float32,float64。
- 输出:
Complex类型的Tensor。
- 异常:
TypeError - 输入的数据类型不是float32或float64之一。
TypeError - 输入的实部与虚部数据类型不一致。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> real = Tensor(np.array([1]), mindspore.float32) >>> imag = Tensor(np.array([2]), mindspore.float32) >>> complex = ops.Complex() >>> output = complex(real, imag) >>> print(output) [1.+2.j]