complex

paddle. complex ( real, imag, name=None ) [源代码]

给定实部和虚部，返回一个复数 Tensor。

参数

  • real (Tensor) - 实部，数据类型为：float32 或 float64。

  • imag (Tensor) - 虚部，数据类型和 real 相同。

  • name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回

输出 Tensor，数据类型是 complex64 或者 complex128，与 real 和 imag 的数值精度一致。

注解

paddle.complex 遵守 broadcasting，如您想了解更多，请参见 cn_user_guide_broadcasting 。

代码示例

import paddle
x = paddle.arange(2, dtype=paddle.float32).unsqueeze(-1)
y = paddle.arange(3, dtype=paddle.float32)
z = paddle.complex(x, y)
print(z)
# Tensor(shape=[2, 3], dtype=complex64, place=Place(cpu), stop_gradient=True,
#        [[0j    , 1j    , 2j    ],
#         [(1+0j), (1+1j), (1+2j)]])