pytorch使用 to 进行类型转换方式


Posted in Python onJanuary 08, 2020

在程序中,有多种方法进行强制类型转换。

本博文将介绍一个非常常用的方法:to()方法。

我们通常使用它来进行GPU和CPU的类型转换,但其实也可以用来进行torch的dtype转换。

常见方法:tensor.to(‘cuda:0')

先看官网介绍:

**Performs Tensor dtype and/or device conversion. A torch.dtype and torch.device are inferred from the arguments of self.to(*args, kwargs).

本文举一个例子,将一个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 进行类型转换方式

可以看到to还是很好用的,尤其是不确定我们的数据类型和device时。

其实pytorch中还有很多其他方法可以这么做,以后会继续介绍。

以上这篇pytorch使用 to 进行类型转换方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python多进程通信Queue、Pipe、Value、Array实例
Nov 21 Python
python获取当前计算机cpu数量的方法
Apr 18 Python
Python实现读写sqlite3数据库并将统计数据写入Excel的方法示例
Aug 07 Python
Python内置模块logging用法实例分析
Feb 12 Python
Python爬虫框架Scrapy实例代码
Mar 04 Python
python计算日期之间的放假日期
Jun 05 Python
如何在Django中设置定时任务的方法示例
Jan 18 Python
python中时间、日期、时间戳的转换的实现方法
Jul 06 Python
Python英文文章词频统计(14份剑桥真题词频统计)
Oct 13 Python
Python识别处理照片中的条形码
Nov 16 Python
Django路由层如何获取正确的url
Jul 15 Python
Github 使用python对copilot做些简单使用测试
Apr 14 Python
Pycharm 2020年最新激活码(亲测有效)
Sep 18 #Python
Pytorch to(device)用法
Jan 08 #Python
pycharm 2019 最新激活方式(pycharm破解、激活)
Sep 22 #Python
Python语言异常处理测试过程解析
Jan 08 #Python
Pytorch在NLP中的简单应用详解
Jan 08 #Python
解析PyCharm Python运行权限问题
Jan 08 #Python
python读取ini配置的类封装代码实例
Jan 08 #Python
You might like
PHP 程序员应该使用的10个组件
2009/10/31 PHP
PHP迭代器实现斐波纳契数列的函数
2013/11/12 PHP
JavaScript中使用replace结合正则实现replaceAll的效果
2010/06/04 Javascript
jQuery 1.5最新版本的改进细节分析
2011/01/19 Javascript
jquery 卷帘效果实现代码(不同方向)
2013/02/05 Javascript
jQuery插件开发详细教程
2014/06/06 Javascript
dreamweaver 8实现Jquery自动提示
2014/12/04 Javascript
javascript实现的登陆遮罩效果汇总
2015/11/09 Javascript
很棒的js选项卡切换效果
2016/07/15 Javascript
Google 地图类型详解及示例代码
2016/08/06 Javascript
D3.js实现折线图的方法详解
2016/09/21 Javascript
对比分析Django的Q查询及AngularJS的Datatables分页插件
2017/02/07 Javascript
利用Chrome DevTools直接调试Node.js和JavaScript的方法详解(并行)
2017/02/16 Javascript
浅谈React高阶组件
2018/03/28 Javascript
详解处理bootstrap4不支持远程静态框问题
2018/07/20 Javascript
浅谈vue父子组件怎么传值
2018/07/21 Javascript
JavaScript时间与时间戳的转换操作实例分析
2018/12/07 Javascript
Vue数据绑定简析小结
2019/05/07 Javascript
Nuxt.js nuxt-link与router-link的区别说明
2020/11/06 Javascript
Python设计模式之代理模式实例
2014/04/26 Python
Python对小数进行除法运算的正确方法示例
2014/08/25 Python
Python的Flask站点中集成xhEditor文本编辑器的教程
2016/06/13 Python
python 时间信息“2018-02-04 18:23:35“ 解析成字典形式的结果代码详解
2018/04/19 Python
python celery分布式任务队列的使用详解
2019/07/08 Python
Python使用Beautiful Soup爬取豆瓣音乐排行榜过程解析
2019/08/15 Python
Django框架 查询Extra功能实现解析
2019/09/04 Python
app内嵌H5 webview 本地缓存问题的解决
2020/10/19 HTML / CSS
中专毕业生自荐信
2013/11/16 职场文书
商务专员岗位职责
2013/11/23 职场文书
优秀团员个人事迹材料
2014/01/29 职场文书
精彩的英文自荐信
2014/01/30 职场文书
美术专业自荐信
2014/07/07 职场文书
德能勤绩廉个人总结
2015/02/14 职场文书
微信早安问候语
2015/11/10 职场文书
创业计划书之废品回收
2019/09/26 职场文书
PyTorch的Debug指南
2021/05/07 Python