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实现目录树生成示例
Mar 28 Python
windows系统中python使用rar命令压缩多个文件夹示例
May 06 Python
Django小白教程之Django用户注册与登录
Apr 22 Python
python解析基于xml格式的日志文件
Feb 25 Python
使用python实现tcp自动重连
Jul 02 Python
Python如何快速上手? 快速掌握一门新语言的方法
Nov 14 Python
Python内置模块turtle绘图详解
Dec 09 Python
pip matplotlib报错equired packages can not be built解决
Jan 06 Python
python3实现带多张图片、附件的邮件发送
Aug 10 Python
python3文件复制、延迟文件复制任务的实现方法
Sep 02 Python
django2.2安装错误最全的解决方案(小结)
Sep 24 Python
利用python+request通过接口实现人员通行记录上传功能
Jan 13 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中修改浏览器的User-Agent来伪装你的浏览器和操作系统
2011/07/29 PHP
php读取纯真ip数据库使用示例
2014/01/26 PHP
PHP中preg_match函数正则匹配的字符串长度问题
2015/05/27 PHP
YII动态模型(动态表名)支持分析
2016/03/29 PHP
PHP网页安全认证的实例详解
2017/09/28 PHP
Laravel访问出错提示:`Warning: require(/vendor/autoload.php): failed to open stream: No such file or di解决方法
2019/04/02 PHP
(跨浏览器基础事件/浏览器检测/判断浏览器)经验代码分享
2013/01/24 Javascript
JavaScript 定时器 SetTimeout之定时刷新窗口和关闭窗口(代码超简单)
2016/02/26 Javascript
jQuery制作网页版选项卡
2016/07/28 Javascript
Bootstrap table的使用方法
2016/11/02 Javascript
Angular的模块化(代码分享)
2016/12/26 Javascript
vue中$refs的用法及作用详解
2018/04/24 Javascript
vue router 用户登陆功能的实例代码
2019/04/24 Javascript
深入理解 JS 垃圾回收
2019/06/03 Javascript
解决vue字符串换行问题(绝对管用)
2020/08/06 Javascript
使用JavaScript实现贪吃蛇游戏
2020/09/29 Javascript
python实现进程间通信简单实例
2014/07/23 Python
Python中endswith()函数的基本使用
2015/04/07 Python
python3调用百度翻译API实现实时翻译
2018/08/16 Python
使用Python编写Prometheus监控的方法
2018/10/15 Python
pycharm: 恢复(reset) 误删文件的方法
2018/10/22 Python
使用python Telnet远程登录执行程序的方法
2019/01/26 Python
使用Django简单编写一个XSS平台的方法步骤
2019/03/25 Python
Django Form 实时从数据库中获取数据的操作方法
2019/07/25 Python
Python调用C语言的实现
2019/07/26 Python
ipython jupyter notebook中显示图像和数学公式实例
2020/04/15 Python
python 解决函数返回return的问题
2020/12/05 Python
杭州联环马网络笔试题面试题
2013/08/04 面试题
大学生专业个人学习的自我评价
2013/10/26 职场文书
药物学专业学生的自我评价
2013/10/27 职场文书
路政管理毕业自荐书范文
2014/02/10 职场文书
出纳员岗位责任制
2014/02/11 职场文书
党的群众路线对照检查材料思想汇报
2014/09/25 职场文书
食品质检员岗位职责
2015/04/08 职场文书
基于HTML十秒做出淘宝页面
2021/10/24 HTML / CSS
什么是css原子化,有什么用?
2022/04/24 HTML / CSS