complex¶
给定实部和虚部,返回一个复数 Tensor.
参数¶
real (Tensor) - 实部,数据类型为:float32 或 float64。
imag (Tensor) - 虚部,数据类型和
real
相同。name (str,可选) - 操作的名称(可选,默认值为None)。更多信息请参见 Name。
返回¶
输出 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.numpy())
# [[0.+0.j 0.+1.j 0.+2.j]
# [1.+0.j 1.+1.j 1.+2.j]]