这篇文章将为大家详细讲解有关怎么在pytorch中使用 to方法转换类型，文章内容质量较高，因此小编分享给大家做个参考，希望大家阅读完这篇文章后对相关知识有一定的了解。
将一个tensor转化成与另一个tensor相同的数据类型和相同GPU或CPU类型
计算
安全
数据库
网络和加速
企业服务
import torch
device = 'cuda:0'
a = torch.zeros(2, 3)
print(type(a))
b = torch.ones(3, 4).to(device)
print(type(b))
c = torch.matmul(a, b)
print(type(c))我们看到这个代码会出错的。因为a和b是不同的device，一个是CPU，一个是GPU，不能运行。
修改如下：
a = a.to(b)
d = torch.matmul(a, b)
print(type(d))
关于怎么在pytorch中使用 to方法转换类型就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
网络异常，请检查网络
