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之入门(五)缩进和选择
May 27 Python
python每隔N秒运行指定函数的方法
Mar 16 Python
举例讲解Python中metaclass元类的创建与使用
Jun 30 Python
python爬虫入门教程--正则表达式完全指南(五)
May 25 Python
遗传算法python版
Mar 19 Python
对Python 3.2 迭代器的next函数实例讲解
Oct 18 Python
Python 判断图像是否读取成功的方法
Jan 26 Python
Python3模拟curl发送post请求操作示例
May 03 Python
Django 静态文件配置过程详解
Jul 23 Python
彻底搞懂 python 中文乱码问题(深入分析)
Feb 28 Python
基于plt.title无法显示中文的快速解决
May 16 Python
Python绘制组合图的示例
Sep 18 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实现的CSS更新类实例
2014/09/22 PHP
实现WordPress主题侧边栏切换功能的PHP脚本详解
2015/12/14 PHP
PHP防止图片盗用(盗链)的方法小结
2016/11/11 PHP
原生JS实现Ajax通过POST方式与PHP进行交互的方法示例
2018/05/12 PHP
wordpress自定义标签云与随机获取标签的方法详解
2019/03/22 PHP
Gambit vs ForZe BO3 第一场 2.13
2021/03/10 DOTA
解决遍历时Array.indexOf产生的性能问题
2012/07/03 Javascript
javascript数字格式化通用类 accounting.js使用
2012/08/24 Javascript
如何在JavaScript中实现私有属性的写类方式(二)
2013/12/04 Javascript
ExtJS4如何给同一个formpanel不同的url
2014/05/02 Javascript
WebSocket+node.js创建即时通信的Web聊天服务器
2016/08/08 Javascript
HTML中setCapture、releaseCapture 使用方法浅析
2016/09/25 Javascript
Javascript实现图片懒加载插件的方法
2016/10/20 Javascript
Bootstrap基本组件学习笔记之导航(10)
2016/12/07 Javascript
简单谈谈gulp-changed插件
2017/02/21 Javascript
详解vue-cli 脚手架项目-package.json
2017/07/04 Javascript
ionic2中使用自动生成器的方法
2018/03/04 Javascript
Angular事件之不同组件间传递数据的方法
2018/11/15 Javascript
Node.js + express实现上传大文件的方法分析【图片、文本文件】
2019/03/14 Javascript
微信小程序位置授权处理方法
2019/06/13 Javascript
javascript触发模拟鼠标点击事件
2019/06/26 Javascript
js实现图片3D轮播效果
2019/09/21 Javascript
python交互式图形编程实例(三)
2017/11/17 Python
python自动12306抢票软件实现代码
2018/02/24 Python
Python 操作 ElasticSearch的完整代码
2019/08/04 Python
印尼最大的在线购物网站:MatahariMall.com
2016/08/26 全球购物
巴西电子产品购物网站:Saldão da Informática
2018/01/09 全球购物
新西兰便宜隐形眼镜购买网站:QUICKLENS New Zealand
2019/03/02 全球购物
澳大利亚领先的武术用品和健身器材供应商:SMAI
2019/03/24 全球购物
Ajax的工作原理
2015/12/04 面试题
我们的节日春节活动方案
2014/08/22 职场文书
群众对十八届四中全会的期盼
2014/10/17 职场文书
社区综治工作汇报
2014/10/27 职场文书
党的群众路线教育实践活动总结材料
2014/10/30 职场文书
街道党风廉政建设调研报告
2015/01/01 职场文书
女性健康讲座主持词
2015/07/04 职场文书