[ torch 参数更多 ]torch.complex¶
torch.complex¶
torch.complex(real, imag, *, out=None)
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
real | real | 实部,数据类型为:float32 或 float64。 |
imag | imag | 虚部,数据类型和 real 相同。 |
out | - | 输出 Tensor。 |
转写示例¶
# PyTorch 写法
torch.complex(a, b, out=out)
# Paddle 写法
paddle.assign(paddle.complex(a, b), out)