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 相关文章推荐
解决windows下Sublime Text 2 运行 PyQt 不显示的方法分享
Jun 18 Python
纯Python开发的nosql数据库CodernityDB介绍和使用实例
Oct 23 Python
Python3搜索及替换文件中文本的方法
May 22 Python
python编程开发之textwrap文本样式处理技巧
Nov 13 Python
Python用threading实现多线程详解
Feb 03 Python
Python对象类型及其运算方法(详解)
Jul 05 Python
Python实战之制作天气查询软件
May 14 Python
Django REST Framework之频率限制的使用
Sep 29 Python
python3实现绘制二维点图
Dec 04 Python
Python利用PyExecJS库执行JS函数的案例分析
Dec 18 Python
python实现简单颜色识别程序
Feb 19 Python
pandas数据拼接的实现示例
Apr 16 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
全国FM电台频率大全 - 13 福建省
2020/03/11 无线电
PHP图片等比例缩放生成缩略图函数分享
2014/06/10 PHP
PHP数组操作类实例
2015/07/11 PHP
PHP封装的MSSql操作类完整实例
2016/05/26 PHP
用jscript启动sqlserver
2007/06/21 Javascript
简单实用的js调试logger组件实现代码
2010/11/20 Javascript
jquery radio 操作代码
2011/03/16 Javascript
设置iframe的document.designMode后仅Firefox中其body.innerHTML为br
2012/02/27 Javascript
div+css+js实现无缝滚动类似marquee无缝滚动兼容firefox
2013/08/29 Javascript
使用indexOf等在JavaScript的数组中进行元素查找和替换
2013/09/18 Javascript
js自动查找select下拉的菜单并选择(示例代码)
2014/02/26 Javascript
js遍历子节点子元素附属性及方法
2014/08/19 Javascript
JQuery实现样式设置、追加、移除与切换的方法
2015/06/11 Javascript
ECMAScript6中Set/WeakSet详解
2015/06/12 Javascript
基于jQuery插件实现点击小图显示大图效果
2016/05/11 Javascript
深入浅析knockout源码分析之订阅
2016/07/12 Javascript
jquery根据一个值来选中select下的option实例代码
2016/08/29 Javascript
利用yarn代替npm管理前端项目模块依赖的方法详解
2017/09/04 Javascript
webpack多入口多出口的实现方法
2018/08/17 Javascript
详解Vue.js在页面加载时执行某个方法
2018/11/20 Javascript
关于vue路由缓存清除在main.js中的设置
2019/11/06 Javascript
jQuery实现简单日历效果
2020/07/05 jQuery
Python之eval()函数危险性浅析
2014/07/03 Python
python中学习K-Means和图片压缩
2017/11/20 Python
基于Python中capitalize()与title()的区别详解
2017/12/09 Python
对matplotlib改变colorbar位置和方向的方法详解
2018/12/13 Python
python判断文件夹内是否存在指定后缀文件的实例
2019/06/10 Python
python验证码图片处理(二值化)
2019/11/01 Python
Python读取分割压缩TXT文本文件实例
2020/02/14 Python
使用CSS3制作饼状旋转载入效果的实例
2015/06/23 HTML / CSS
英国拳击装备购物网站:RDX Sports
2018/01/23 全球购物
药剂专业学生求职信范文
2013/12/28 职场文书
副董事长岗位职责
2014/04/02 职场文书
高考1977观后感
2015/06/04 职场文书
水知道答案观后感
2015/06/08 职场文书
Python GUI编程之tkinter 关于 ttkbootstrap 的使用详解
2022/03/03 Python