complex

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

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

参数

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

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

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

返回

输出 Tensor,数据类型是 complex64 或者 complex128,与 realimag 的数值精度一致。

注解

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)]])