Pytorch中的variable, tensor与numpy相互转化的方法

更新时间：2019年10月10日 10:05:55 作者：pengge0433

这篇文章主要介绍了Pytorch中的variable, tensor与numpy相互转化的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

在使用pytorch作为深度学习的框架时，经常会遇到变量variable、张量tensor与矩阵numpy的类型的相互转化的问题，本章结合这实际图像对此转化方法进行实现。

1.加载需要用到的模块

import torch
from torch.autograd import Variable
import matplotlib.pyplot as plt
import matplotlib.image as mpimg

2.显示图片与图片中的一部分区域

test_img = mpimg.imread('example1.jpg')
i_x = 20
i_y = 85
sub_img = test_img[i_y:i_y + 100,i_x:i_x + 100,:] #numpy类型

3.将numpy矩阵转换为Tensor张量

sub_ts = torch.from_numpy(sub_img)  #sub_img为numpy类型

4.将Tensor张量转化为numpy矩阵

sub_np1 = sub_ts.numpy()       #sub_ts为tensor张量

5.将numpy转换为Variable

sub_va = Variable(torch.from_numpy(sub_img))

6.将Variable张量转化为numpy

sub_np2 = sub_va.data.numpy()

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

在 Python 中接管键盘中断信号的实现方法
要使用信号，我们需用导入 Python 的signal库。然后自定义一个信号回调函数，当 Python 收到某个信号时，调用这个函数。，下面通过实例代码给大家介绍在 Python 中接管键盘中断信号,需要的朋友可以参考下
2020-02-02
Python while 循环使用的简单实例
下面小编就为大家带来一篇Python while 循环使用的简单实例。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2016-06-06
浅析python中的del用法
python中的del用法比较特殊，新手学习往往产生误解，弄清del的用法，可以帮助深入理解python的内存方面的问题。这篇文章主要介绍了python中的del用法,需要的朋友可以参考下
2020-09-09
PyCharm2019.3永久激活破解详细图文教程，亲测可用（不定期更新）
这篇文章主要介绍了PyCharm2019.3最新激活码(注册码)破解永久版详细图文教程的相关资料,亲测可用，需要的朋友可以参考下
2020-10-10
Python之虚拟环境virtualenv,pipreqs生成项目依赖第三方包的方法
今天小编就为大家分享一篇Python之虚拟环境virtualenv,pipreqs生成项目依赖第三方包的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-07-07
pyqt5+opencv 实现读取视频数据的方法
这篇文章主要介绍了pyqt5+opencv 实现读取视频数据的方法，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2022-01-01
python全局变量引用与修改过程解析
这篇文章主要介绍了python全局变量引用与修改过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-01-01
Python、 Pycharm、Django安装详细教程(图文)
这篇文章主要介绍了Python、 Pycharm、Django安装详细教程，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-04-04
在pycharm中设置显示行数的方法
今天小编就为大家分享一篇在pycharm中设置显示行数的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-01-01
Python读取xlsx文件报错:xlrd.biffh.XLRDError: Excel xlsx file;no
这篇文章主要给大家介绍了关于Python库xlrd中的xlrd.open_workbook()函数读取xlsx文件报错:xlrd.biffh.XLRDError: Excel xlsx file;not supported问题解决的相关资料,文中通过图文介绍的非常详细,需要的朋友可以参考下
2022-08-08