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 爬虫的工具列表大全
Jan 31 Python
关于python写入文件自动换行的问题
Jun 23 Python
Python爬虫框架Scrapy常用命令总结
Jul 26 Python
Python函数any()和all()的用法及区别介绍
Sep 14 Python
python 不同方式读取文件速度不同的实例
Nov 09 Python
解析Python的缩进规则的使用
Jan 16 Python
聊聊python里如何用Borg pattern实现的单例模式
Jun 06 Python
Django中的cookie和session
Aug 27 Python
python创建子类的方法分析
Nov 28 Python
Python3基于print打印带颜色字符串
Jul 06 Python
Python3使用tesserocr识别字母数字验证码的实现
Jan 29 Python
实操Python爬取觅知网素材图片示例
Nov 27 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利用MySQL保存session的实现思路及示例代码
2014/09/09 PHP
PHP判断网络文件是否存在的方法
2015/03/12 PHP
php猜单词游戏
2015/09/29 PHP
PHP基于cookie与session统计网站访问量并输出显示的方法
2016/01/15 PHP
提高Laravel应用性能方法详解
2019/06/24 PHP
Laravel定时任务的每秒执行代码
2019/10/22 PHP
IE图片缓存document.execCommand("BackgroundImageCache",false,true)
2011/03/01 Javascript
JavaScript中的onerror事件概述及使用
2013/04/01 Javascript
javascript格式化指定日期对象的方法
2015/04/21 Javascript
jQuery模拟360浏览器切屏效果幻灯片(附demo源码下载)
2016/01/29 Javascript
jQuery使用each方法与for语句遍历数组示例
2016/06/16 Javascript
详解Vue路由开启keep-alive时的注意点
2017/06/20 Javascript
详解开源的JavaScript插件化框架MinimaJS
2017/10/26 Javascript
第一个Vue插件从封装到发布
2017/11/22 Javascript
探索Vue高阶组件的使用
2018/01/08 Javascript
javascript面向对象创建对象的方式小结
2019/07/29 Javascript
vue实现pdf文档在线预览功能
2019/11/26 Javascript
JavaScript 实现自己的安卓手机自动化工具脚本(推荐)
2020/05/13 Javascript
[36:52]DOTA2真视界:基辅特锦赛总决赛
2017/05/21 DOTA
python实现定时同步本机与北京时间的方法
2015/03/24 Python
修改Python的pyxmpp2中的主循环使其提高性能
2015/04/24 Python
用python记录运行pid,并在需要时kill掉它们的实例
2017/01/16 Python
HTML5 File API改善网页上传功能
2009/08/19 HTML / CSS
html5 Canvas画图教程(9)—canvas中画出矩形和圆形
2013/01/09 HTML / CSS
潘多拉珠宝俄罗斯官方网上商店:PANDORA俄罗斯
2020/09/22 全球购物
俄罗斯连接商品和买家的在线平台:goods.ru
2020/11/30 全球购物
环境科学专业个人求职的自我评价
2013/11/28 职场文书
学生请假条
2014/04/11 职场文书
环境保护建议书
2014/08/26 职场文书
防灾减灾日活动总结
2014/08/26 职场文书
党员群众路线剖析材料
2014/10/08 职场文书
党的群众路线教育实践活动专题组织生活会发言材料
2014/10/17 职场文书
网络营销计划书
2015/01/17 职场文书
教师节随笔
2015/08/15 职场文书
制作能在nginx和IIS中使用的ssl证书
2021/06/21 Servers
Windows server 2012 R2 安装IIS服务器
2022/04/29 Servers