怎么在pytorch中使用to方法转换类型-创新互联

这篇文章将为大家详细讲解有关怎么在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方法转换类型就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。

网页标题：怎么在pytorch中使用to方法转换类型-创新互联
文章地址：https://www.cdcxhl.com/article20/hgpjo.html

成都网站建设公司_创新互联，为您提供做网站、App设计、服务器托管、用户体验、网站建设、网站设计公司

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源：创新互联

猜你还喜欢下面的内容