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从ftp下载数据保存实例
Nov 20 Python
在Python编程过程中用单元测试法调试代码的介绍
Apr 02 Python
Python中关键字nonlocal和global的声明与解析
Mar 12 Python
深入理解Python中的内置常量
May 20 Python
神经网络相关之基础概念的讲解
Dec 29 Python
Python常用模块之requests模块用法分析
May 15 Python
python 处理微信对账单数据的实例代码
Jul 19 Python
python logging模块书写日志以及日志分割详解
Jul 22 Python
浅谈Python中的异常和JSON读写数据的实现
Feb 27 Python
Python网络爬虫信息提取mooc代码实例
Mar 06 Python
python中if及if-else如何使用
Jun 02 Python
python定义具名元组实例操作
Feb 28 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版(5)
2006/10/09 PHP
php将session放入memcached的设置方法
2014/02/14 PHP
爱恋千雪-US-AscII加密解密工具(网页加密)下载
2007/06/06 Javascript
JavaScript学习历程和心得小结
2010/08/16 Javascript
3款实用的在线JS代码工具(国外)
2012/03/15 Javascript
Javascript遍历Html Table示例(包括内容和属性值)
2014/07/08 Javascript
详解AngularJS中的作用域
2015/06/17 Javascript
jQuery判断浏览器并动态调整select宽度的方法
2016/03/02 Javascript
chrome浏览器如何断点调试异步加载的JS
2016/09/05 Javascript
JavaScript轻松创建级联函数的方法示例
2017/02/10 Javascript
easyui简介_动力节点Java学院整理
2017/07/14 Javascript
浅谈Vue数据响应思路之数组
2018/11/06 Javascript
Nodejs libuv运行原理详解
2019/08/21 NodeJs
NUXT SSR初级入门笔记(小结)
2019/12/16 Javascript
在Linux下使用Python的matplotlib绘制数据图的教程
2015/06/11 Python
PyCharm鼠标右键不显示Run unittest的解决方法
2018/11/30 Python
Python从入门到精通之环境搭建教程图解
2019/09/26 Python
python实现图片二值化及灰度处理方式
2019/12/07 Python
什么是Python中的匿名函数
2020/06/02 Python
使用HTML5进行SVG矢量图形绘制的入门教程
2016/02/19 HTML / CSS
Oakley官网:运动太阳镜、雪镜和服装
2016/09/30 全球购物
戴尔新加坡官网:Dell Singapore
2020/12/13 全球购物
英国顶尖手表珠宝品牌独家授权经销商:HS Johnson
2020/10/28 全球购物
职业教育毕业生求职信
2013/11/09 职场文书
学校食堂采购员岗位职责
2013/12/05 职场文书
计算机相关的自我评价
2014/01/15 职场文书
自动化专业职业生涯规划书范文
2014/01/16 职场文书
父亲八十大寿答谢词
2014/01/23 职场文书
军人违纪检讨书
2014/02/04 职场文书
楼面经理岗位职责范本
2014/02/18 职场文书
党支部党的群众路线对照检查材料
2014/09/24 职场文书
余世维讲座观后感
2015/06/11 职场文书
小学班主任教育随笔
2015/08/15 职场文书
小组组名及励志口号
2015/12/24 职场文书
学校中层领导培训心得体会
2016/01/11 职场文书
一篇文章弄懂MySQL查询语句的执行过程
2021/05/07 MySQL