这个屌丝很懒，什么也没留下！

⽤方法 to() 可以将 Tensor 在CPU和GPU（需要硬件支持）之间相互移动_torch tensor to cpu

作者：小丑西瓜9 | 2024-06-14 03:06:56

torch tensor to cpu


# 以下代码只有在PyTorch GPU版本上才会执⾏行行
if torch.cuda.is_available():
device = torch.device("cuda") # GPU
y = torch.ones_like(x, device=device) # 直接创建一个在GPU上的Tensor
x = x.to(device) # 等价于 .to("cuda")
z = x + y
print(z)
print(z.to("cpu", torch.double)) # to()还可以同时更改数据类型

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小丑西瓜9/article/detail/716013