PyTorch|view(),改变张量维度

在构建自己的网络时,了解数据经过每个层后的形状变化是必须的,否则,网络大概率会出现问题。PyToch张量有一个方法,叫做view(),使用这个方法,我们可以很容易的对张量的形状进行改变,从而符合网络的输入要求

view()的基本用法很简单,只需传入想要的形状即可,

就像这样:

复制代码
import torchT1=torch.arange(0,16)print(T1)print(T1.size())
tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15])torch.Size([16])
a=T1.view(4,4)b=T1.view(2,8)print(a)print(a.size())
print(b)print(b.size())
tensor([[ 0,  1,  2,  3],        [ 4,  5,  6,  7],        [ 8,  9, 10, 11],        [12, 13, 14, 15]])torch.Size([4, 4])
tensor([[ 0,  1,  2,  3,  4,  5,  6,  7],        [ 8,  9, 10, 11, 12, 13, 14, 15]])torch.Size([2, 8])

当然,在看一些代码时,往往会发现view()中有参数-1。看下面几个例子​​​​​​​

复制代码
c=T1.view(-1,4,4)print(c)print(c.size())tensor([[[ 0,  1,  2,  3],         [ 4,  5,  6,  7],         [ 8,  9, 10, 11],         [12, 13, 14, 15]]])torch.Size([1, 4, 4]​​​​​​​

d=T1.view(-1,8)print(d)print(d.size())tensor([[ 0,  1,  2,  3,  4,  5,  6,  7],        [ 8,  9, 10, 11, 12, 13, 14, 15]])torch.Size([2, 8])​​​​​​​

e=T1.view(8,-1)print(e)print(e.size())tensor([[ 0,  1],        [ 2,  3],        [ 4,  5],        [ 6,  7],        [ 8,  9],        [10, 11],        [12, 13],        [14, 15]])torch.Size([8, 2])

通过上述几个例子,view中一个参数定为-1,代表自动调整这个维度上的元素个数,同时保证元素的总个数不变当然,我们也注意到了这个细节:传入几个参数,代表结果有几个维度

对于这篇文章:张量扁平化------CNN的Flatten操作

保持batch轴不变,展平其它轴,我们采用的是flatten()方法,这里我们用view()方法。​​​​​​​

复制代码
t1=torch.rand([5,3,128,128])print(t1.size())
torch.Size([5, 3, 128, 128])​​​​​​​

t2=t1.view(5,-1)print(t2)print(t2.size())

结果符合我们的预期!

相关推荐
PingCAP14 分钟前
TiDB 亮相宜昌“医院‘云数智’技术实践研讨及成果展示交流会”,探讨国产化 + AI 背景下的数据库新趋势
数据库·人工智能·tidb
文弱_书生14 分钟前
再谈图像处理中的傅里叶变换
图像处理·人工智能·傅里叶变换
钡铼技术物联网关17 分钟前
ARM边缘计算时代:BLIoTLink如何打通设备互联任督二脉
arm开发·人工智能·边缘计算
mywpython21 分钟前
mac 最新的chrome版本配置selenium的方式
chrome·python·selenium·macos
小李独爱秋23 分钟前
机器学习开发全流程详解:从数据到部署的完整指南
人工智能·机器学习
闲人编程25 分钟前
形态学操作(腐蚀/膨胀/开闭运算)
python·opencv·图像识别
A_ugust__25 分钟前
vue3项目使用 python +flask 打包成桌面应用
开发语言·python·flask
Dovis(誓平步青云)26 分钟前
深挖 DeepSeek 隐藏玩法·智能炼金术2.0版本
人工智能·深度学习·机器学习·数据挖掘·服务发现·智慧城市
软件测试曦曦26 分钟前
如何使用Python自动化测试工具Selenium进行网页自动化?
自动化测试·软件测试·python·功能测试·测试工具·程序人生·自动化
陈明勇28 分钟前
一文掌握 MCP 上下文协议:从理论到实践
人工智能·后端·mcp