pytorch每日一学26(torch.complex())生成复数tensor

最新推荐文章于 2026-05-06 05:42:55 发布

原创最新推荐文章于 2026-05-06 05:42:55 发布 · 1.5w 阅读

18 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python #深度学习 #pytorch #神经网络 #机器学习

收录于

pytorch每日一学专栏收录该内容

49 篇文章

订阅专栏

该博客介绍了如何使用PyTorch的torch.complex方法创建复数Tensor。通过指定实部参数`real`和虚部参数`imag`，可以生成复数Tensor。当输入为float32类型时，生成的复数Tensor为complex64类型。示例中展示了创建并打印复数Tensor的过程。

该文章已生成可运行项目，

第26个方法

torch.complex(real, imag, *, out=None) → Tensor

此方法其实额前面讲过的一些方法有很大的关系，此方法为创建复数tensor的方法，其中real为实部，imag为虚部，real和imag必须位数相同，如果real和imag同为float32那么生成的complex就为complex64,。
参数介绍：
real：生成的复数中的实部。
imag：生成的复数中的虚部。
out：输出的tensor。

使用方法如下：

>>> real = torch.tensor([1, 2], dtype=torch.float32)
>>> imag = torch.tensor([3, 4], dtype=torch.float32)
>>> z = torch.complex(real, imag)
>>> z
tensor([(1.+3.j), (2.+4.j)])
>>> z.dtype
torch.complex64