怎么在pytorch中使用,方法转换类型 - 行业资讯 - 肥雀云

　　介绍

这篇文章将为大家详细讲解有关怎么在pytorch中使用,方法转换类型,文章内容质量较高,因此小编分享给大家做个参考,希望大家阅读完这篇文章后对相关知识有一定的了解。

将一个张量转化成与另一个张量相同的数据类型和相同GPU或CPU类型

import 火炬　　　　时间=device & # 39; cuda: 0 & # 39; 　　　　时间=a torch.zeros (2, 3) 　　print(类型(a)) 　　　　时间=b torch.ones(3, 4),(设备) 　　打印((b)型) 　　　　时间=c torch.matmul (a, b) 　　打印(类型(c))

我们看到这个代码会出错的。因为a和b是不同的装置,一个是CPU、一个是GPU,不能运行。

修改如下:

a =,占据(b) 　　时间=d torch.matmul (a, b) 　　打印(类型(d))

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

关于怎么在pytorch中使用,方法转换类型就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看的到。