as_complex¶
将实数 Tensor 转为复数 Tensor。
输入 Tensor 的数据类型是 'float32' 或者 'float64',输出 Tensor 的数据类型相应为 'complex64' 或者 'complex128'。
输入 Tensor 的形状是 (*, 2)
(其中 *
表示任意形状),亦即,输入的最后一维的大小必须是 2,这对应着复数的实部和虚部。输出 Tensor 的形状是 (*,)
。
返回¶
输出 Tensor,数据类型是 'complex64' 或 'complex128',与 x
的数值精度一致。
代码示例¶
import paddle
x = paddle.arange(12, dtype=paddle.float32).reshape([2, 3, 2])
y = paddle.as_complex(x)
print(y)
# Tensor(shape=[2, 3], dtype=complex64, place=Place(gpu:0), stop_gradient=True,
# [[1j , (2+3j) , (4+5j) ],
# [(6+7j) , (8+9j) , (10+11j)]])