Pytorch中view函数实例讲解
一、函数简介
Pytorch中的view函数主要用于Tensor维度的重构，即返回一个有相同数据但不同维度的Tensor。
根据上面的描述可知，view函数的操作对象应该是Tensor类型。如果不是Tensor类型，可以通过tensor = torch.tensor(data)来转换。
二、实例讲解
▶view(参数a,参数b,…)，其中，总的参数个数表示将张量重构后的维度。
import torch temp = [1,2,3,4,5,6] # temp的类型为list，非Tensor temp = torch.tensor(temp) # 将temp由list类型转为Tensor类型 print(temp) # torch.Size([6]) print(temp.view(2,3)) # 将temp的维度改为2*3 print(temp.view(2,3,1)) # 将temp的维度改为2*3*1 print(temp.view(2,3,1,1)) # 更多的维度也没有问题，只要保证维度改变前后的元素个数相同就行，即2*3*1*1=6
▶view(参数a,参数b,…)，其中，如果某个参数为-1，则表示该维度取决于其它维度，由Pytorch自己补充。
import torch
temp = [[11,12,13,14,15,16],
[21,22,23,24,25,26]]
temp = torch.tensor(temp)
print(temp)
# torch.Size([2, 6])
print(temp.view(3,-1,2))
# 这里的-1表示该维度取决于其它维度，即等于(2*6)÷3÷2=2
# torch.Size([3, 2, 2])
▶view(-1)表示将Tensor转为一维Tensor。
import torch temp = [1,2,3,4,5,6] # temp的类型为list，非Tensor temp = torch.tensor(temp) # 将temp由list类型转为Tensor类型 print(temp) # 本身就是一维张量 print(temp.view(-1)) # 因此，转变后还是一维，没什么变换 temp1 = torch.tensor([[1,2,3],[4,5,6]]) print(temp1) # torch.Size([2, 3]) print(temp1.view(-1)) # 多维张量转为一维张量
附：Pytorch中的view函数需要注意的点
一、函数说明
Pytorch中的view函数主要用于张量tensor的维度修改，通过view里面的参数，可以修改成任意维度的张量，前提是修改后的张量大小和原张量大小一致，比如45的张量，可以修改成225，但是不能修改成32*3。
二、数据共享
通过view修改维度以后的张量，和原张量是共享底层数据的，也就是说，如果你修改了一个张量中的数据，两个张量数据都会改变。比如原张量V1=torch.zeros(4,5)，修改后的张量V2=V1.view(2,2,5)，如果让V2[0][0][0]=1，则V1[0][0]也等于1。
三、参数
view的参数如果有-1，则表示这个维度的大小由其他位置决定，只能用剩下的数据维度。
还是V1=torch.zeros(4,5)，V3=V1.view(2,-1,5)，则V3的第二维度大小就是2了。每次只能有一个默认的-1维度，不然没法计算准确的维度大小。
四、维度变化
还是V1=torch.zeros(4,5)，V4=V1.view(20,1)表示的是二维张量20*1，而V5=V1.view(20)，或V5=V1.view(20，)表示的是一维张量，长度为20。在训练模型的时候需要注意，因为这两种方式都能进入全连接层或者线性神经网络，只不过出来的结果会对应输入的维度，在计算误差的时候，对目标的处理就不一样了，需要相同维度，也就是在loss = nn.xxxxxLoss(output, label)时，要注意ouput和label的维度是否一致。
总结
到此这篇关于Pytorch中view函数实例讲解的文章就介绍到这了,更多相关Pytorch view函数内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了详细一文带你分清Python中的模块、包和库,Python 模块(Module),是一个 Python 文件,以 .py 结尾,包含了 Python 对象定义和Python语句,模块能定义函数,类和变量,模块也能包含可执行的代码,需要的朋友可以参考下2023-08-08
这篇文章主要介绍了10款最佳Python开发工具推荐,每一款都是神器，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2020-10-10
PIL是专为Python语言设计的图像处理库,它涵盖了广泛的图像处理功能,如图像的加载、保存、编辑,以及执行多样化的图像处理任务,本文给大家介绍了Python使用PIL将图片或GIF转为字符画的方法,需要的朋友可以参考下2025-03-03
这篇文章主要介绍了Python数据结构与算法之图结构（Graph）,结合实例形式分析了图结构的概念、原理、使用方法及相关操作技巧,需要的朋友可以参考下2017-09-09
这篇文章主要为大家详细介绍一下Python的一个神操作，那就是实现乱序文件重新命名编号功能，文中的示例代码讲解详细，感兴趣的可以尝试一下2022-08-08
这篇文章主要给大家介绍了关于Python爬虫利用lxml模块爬取豆瓣读书排行榜的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2019-04-04
Python编程中内置的NotImplemented类型的用法
这篇文章主要介绍了Python编程中内置的NotImplemented类型的用法，NotImplemented 是Python在内置命名空间中的六个常数之一，下文更多详细内容需要的小伙伴可以参考一下2022-03-03
下面小编就为大家带来一篇Python中创建字典的几种方法总结(推荐)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-04-04
这篇文章主要介绍了解决Jupyter-notebook不弹出默认浏览器的问题,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-03-03
这篇文章主要介绍了Python基本知识点总结,Python基本知识,Python字符串,Python函数,Python模块,Python创建类,Python类的继承,Python重载,Python单下划线说明,Python双下划线说明,Python头尾双下划线说明,需要的朋友可以参考下2022-04-04
最新评论