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的Scrapy爬虫框架使用代理进行采集的方法
Feb 18 Python
python计算auc指标实例
Jul 13 Python
python3 读写文件换行符的方法
Apr 09 Python
python实现多进程代码示例
Oct 31 Python
实例介绍Python中整型
Feb 11 Python
20行python代码的入门级小游戏的详解
May 05 Python
Python 20行简单实现有道在线翻译的详解
May 15 Python
python实现梯度法 python最速下降法
Mar 24 Python
基于python爬取链家二手房信息代码示例
Oct 21 Python
基于pycharm 项目和项目文件命名规则的介绍
Jan 15 Python
python基于爬虫+django,打造个性化API接口
Jan 21 Python
Django实现翻页的示例代码
May 24 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的FTP学习(二)[转自奥索]
2006/10/09 PHP
第三节--定义一个类
2006/11/16 PHP
Laravel与CI框架中截取字符串函数
2016/05/08 PHP
PHP生成(支持多模板)二维码海报代码
2018/04/30 PHP
JavaScript词法作用域与调用对象深入理解
2012/11/29 Javascript
JQuery AJAX 中文乱码问题解决
2013/06/05 Javascript
jquery通过ajax加载一段文本内容的方法
2015/01/15 Javascript
js给selected添加options的方法
2015/05/06 Javascript
AngularJS基础知识笔记之表格
2015/05/10 Javascript
js实现简洁的滑动门菜单(选项卡)效果代码
2015/09/04 Javascript
javascript设置文本框光标的方法实例小结
2016/11/04 Javascript
Dropzone.js实现文件拖拽上传功能(附源码下载)
2016/11/22 Javascript
AngularJs验证重复密码的方法(两种)
2016/11/25 Javascript
JavaScript 函数节流详解及方法总结
2017/02/09 Javascript
js将当前时间格式化为 年-月-日 时:分:秒的实现代码
2018/01/20 Javascript
js中Array对象的常用遍历方法详解
2019/01/17 Javascript
ES6入门教程之变量的解构赋值详解
2019/04/13 Javascript
详解关于表格合并span-method方法的补充(表格数据由后台动态返回)
2019/05/21 Javascript
layer.prompt输入层的例子
2019/09/24 Javascript
vue内置组件component--通过is属性动态渲染组件操作
2020/07/28 Javascript
关于vue的列表图片选中打钩操作
2020/09/09 Javascript
[00:14]PWL:老朋友Mushi拍VLOG与中国玩家问好
2020/11/04 DOTA
python实现根据用户输入从电影网站获取影片信息的方法
2015/04/07 Python
使用Python操作Elasticsearch数据索引的教程
2015/04/08 Python
Python3实现Web网页图片下载
2016/01/28 Python
python WindowsError的错误代码详解
2017/07/23 Python
Python实现的爬取网易动态评论操作示例
2018/06/06 Python
删除python pandas.DataFrame 的多重index实例
2018/06/08 Python
对Python中plt的画图函数详解
2018/11/07 Python
python sklearn库实现简单逻辑回归的实例代码
2019/07/01 Python
opencv python 图像轮廓/检测轮廓/绘制轮廓的方法
2019/07/03 Python
解决python对齐错误的方法
2020/07/16 Python
Athleta官网:购买女士瑜伽服、技术运动服和休闲运动服
2020/11/12 全球购物
幼儿园园长岗位职责
2013/11/26 职场文书
八项规定整改措施
2014/02/12 职场文书
2015年幼儿园元旦亲子活动方案
2014/12/09 职场文书