这篇文章将为大家详细讲解有关怎么在pytorch中使用 to方法转换类型，文章内容质量较高，因此小编分享给大家做个参考，希望大家阅读完这篇文章后对相关知识有一定的了解。
成都创新互联成立与2013年，先为嘉荫等服务建站，嘉荫等地企业，进行企业商务咨询服务。为嘉荫企业网站制作PC+手机+微官网三网同步一站式服务解决您的所有建站问题。
将一个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方法转换类型就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。
网站题目：怎么在pytorch中使用to方法转换类型-创新互联
分享网址：
http://cdkjz.cn/article/hgpjo.html
