pytorch tensor/数据类型转化

最新推荐文章于 2025-01-15 11:20:35 发布

冬日and暖阳

最新推荐文章于 2025-01-15 11:20:35 发布

阅读量625

点赞数

分类专栏： python 文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/qq_29007291/article/details/128314234

版权

python 专栏收录该内容

8 篇文章

订阅专栏

pytorch 怎么进行数据类型的转化呢？

https://pytorch.org/docs/stable/generated/torch.Tensor.to.html

使用类方法 to，可以转化数据类型和数据postion，即可以把tensor在CPU和GPU之前进行转化

tensor = torch.randn(2, 2)  # Initially dtype=float32, device=cpu
tensor.to(torch.float64)

cuda0 = torch.device('cuda:0')
tensor.to(cuda0)

tensor.to(cuda0, dtype=torch.float64)

other = torch.randn((), dtype=torch.float64, device=cuda0)
tensor.to(other, non_blocking=True)