怎么在pytorch中使用 to方法转换类型

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