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中optionParser模块的使用方法实例教程
Aug 29 Python
跟老齐学Python之用Python计算
Sep 12 Python
python 安装virtualenv和virtualenvwrapper的方法
Jan 13 Python
Python中几种导入模块的方式总结
Apr 27 Python
python创建文件时去掉非法字符的方法
Oct 31 Python
Python3利用print输出带颜色的彩色字体示例代码
Apr 08 Python
简单了解python的内存管理机制
Jul 08 Python
关于Python 常用获取元素 Driver 总结
Nov 24 Python
pycharm部署、配置anaconda环境的教程
Mar 24 Python
解析python 中/ 和 % 和 //(地板除)
Jun 28 Python
解决python 虚拟环境删除包无法加载的问题
Jul 13 Python
cookies应对python反爬虫知识点详解
Nov 25 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
收音机指标测试方法及仪器
2021/03/01 无线电
浅谈ThinkPHP的URL重写
2014/11/25 PHP
php+Mysqli利用事务处理转账问题实例
2015/02/11 PHP
使用Huagepage和PGO来提升PHP7的执行性能
2015/11/30 PHP
如何解决PHP获取不到SESSION信息之一般情况
2019/10/10 PHP
PHP扩展安装方法步骤解析
2020/11/24 PHP
深入理解JavaScript系列(16) 闭包(Closures)
2012/04/12 Javascript
javascript特殊用法示例介绍
2013/11/29 Javascript
JQuery插件iScroll实现下拉刷新,滚动翻页特效
2014/06/22 Javascript
点击button获取text内容并改变样式的js实现
2014/09/09 Javascript
拥Bootstrap入怀——导航栏篇
2016/05/30 Javascript
jquery checkbox的相关操作总结
2016/10/17 Javascript
使用vue实现点击按钮滑出面板的实现代码
2017/01/10 Javascript
javascript实现的图片预览功能
2017/03/25 Javascript
实例讲解javascript实现异步图片上传方法
2017/12/05 Javascript
javaScript手机号码校验工具类PhoneUtils详解
2017/12/08 Javascript
vue中rem的配置的方法示例
2018/08/30 Javascript
Vue-不允许嵌套式的渲染方法
2018/09/13 Javascript
Python格式化压缩后的JS文件的方法
2015/03/05 Python
Python装饰器使用示例及实际应用例子
2015/03/06 Python
详解 Python 与文件对象共事的实例
2017/09/11 Python
python 执行shell命令并将结果保存的实例
2018/05/11 Python
python3.6使用urllib完成下载的实例
2018/12/19 Python
Python语言进阶知识点总结
2019/05/28 Python
Python基于机器学习方法实现的电影推荐系统实例详解
2019/06/25 Python
pytorch多GPU并行运算的实现
2019/09/27 Python
pytorch dataloader 取batch_size时候出现bug的解决方式
2020/02/20 Python
python获取栅格点和面值的实现
2020/03/10 Python
米兰网婚纱礼服法国网上商店:Milanoo法国
2016/08/20 全球购物
serialVersionUID具有什么样的特征
2014/02/20 面试题
J2EE面试题大全
2016/08/06 面试题
大学生个人求职信范文
2013/09/21 职场文书
道德模范先进事迹
2014/02/14 职场文书
机械制造专业毕业生求职信
2014/03/02 职场文书
2014最新离职证明范本
2014/09/12 职场文书
2014年银行个人工作总结
2014/12/05 职场文书