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中numpy包使用教程之数组和相关操作详解
这篇文章主要给大家介绍了关于python中numpy包的使用教程，包含数组和相关操作等内容，文中通过示例代码介绍的非常详细，对大家具有一定的参考学习价值，需要的朋友们下面来跟着小编一起学习学习吧。
2017-07-07
python实现对变位词的判断方法
这篇文章主要为大家详细介绍了python实现对变位词的判断方法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2020-04-04
python解析xml文件实例分享
这篇文章主要介绍了python解析XML文件的方法，大家参考使用吧
2013-12-12
Python解析最简单的验证码
最近在学python，正好遇到学校需要选宿舍，就用python写了一个抢宿舍的软件。下面通过本文给大家介绍python解析最简单的验证码，对python解析验证码相关知识感兴趣的朋友一起学习吧
2016-01-01
python中Matplotlib实现绘制3D图的示例代码
本篇文章主要介绍了python中Matplotlib实现绘制3D图的示例代码，具有一定的参考价值，有兴趣的可以了解一下
2017-09-09
python实现分页效果
这篇文章主要为大家详细介绍了python实现分页效果，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2017-10-10
Python之inspect模块实现获取加载模块路径的方法
今天小编就为大家分享一篇Python之inspect模块实现获取加载模块路径的方法，具有很好的价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-10-10
keras实现多GPU或指定GPU的使用介绍
这篇文章主要介绍了keras实现多GPU或指定GPU的使用介绍，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-06-06
python中如何实现链式调用
这篇文章主要介绍了python中如何实现链式调用，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下
2021-03-03
python匹配两个短语之间的字符实例
今天小编就为大家分享一篇python匹配两个短语之间的字符实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-12-12