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

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> 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]