torch.Tensor.to的用法

00000cj

于 2022-07-27 15:49:18 发布

阅读量3.7k

点赞数 4

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

本文链接：https://blog.csdn.net/ooooocj/article/details/126016499

版权

Pytorch 专栏收录该内容

11 篇文章

订阅专栏

本文介绍了如何使用PyTorch操作Tensor对象，包括更改数据类型torch.dtype（如从float32转为int64）和设备torch.device（如从CPU迁移到GPU）。通过实例演示了如何查看数据类型和设备，并展示了如何将Tensor移动到特定CUDA设备上。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

更改数据类型torch.dtype或者设备torch.device

x = torch.Tensor([1, 2, 3])
print(x.dtype)  # torch.float32
x = x.to(torch.int64)
print(x.dtype)  # torch.int64

print(x.device)  # device(type='cpu')
print(x)  # tensor([1, 2, 3])
cuda0 = torch.device('cuda:0')
x = x.to(cuda0)
print(x.device)  # device(type='cuda', index=0)
print(x)  # tensor([1, 2, 3], device='cuda:0')

参考

torch.Tensor.to — PyTorch 1.12 documentation