complex

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

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

参数

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

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

  • name (str,可选) - 操作的名称(可选,默认值为None)。更多信息请参见 Name

返回

输出 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.numpy())

# [[0.+0.j 0.+1.j 0.+2.j]
#  [1.+0.j 1.+1.j 1.+2.j]]