重新塑造数据--PyTorch中的“view”函数详解

PyTorch中的view函数:张量形状变换与神经网络应用
本文介绍了PyTorch中的view函数,用于改变张量形状而保持数据不变,适用于深度学习任务如CNN。它能压缩、展开、增加或减少维度,并与unsqueeze和squeeze配合使用。尤其在神经网络中处理多维输出时,view的灵活性和内存效率至关重要。

目录

一、概述

二、view函数多种应用

1、改变张量的形状

2、压缩和展开张量

3、增加或减少维度

4、与unsqueeze和squeeze结合使用

三、特殊使用:参数-1

四、总结


一、概述

view 函数是PyTorch中用于改变张量(tensor)形状的方法之一。它允许您重新排列张量的维度和大小,但不改变张量中的数据,类似于numpy中的reshape,重新定义矩阵的形状。

view 操作只是改变张量的元数据,而不会复制张量中的数据。这意味着它非常高效,不会占用额外的内存,因为它与原始张量共享相同的数据存储。在深度学习中,view 函数通常用于整理数据以适应神经网络的输入或输出要求。例如,卷积神经网络(CNN)中,通常需要将图像数据整理为特定的形状,以便输入到网络中。

二、view函数多种应用

1、改变张量的形状
x = torch.arange(12).reshape(3, 4) 
y = x.view(2, 6)

从一个多维张量变成另一个多维张量,只要新形状的元素数量与原始张量的元素数量一致。该案例创造一个3*4的张量(矩阵),通过view函数将该张量修改2*6,但总的大小不改变。

2、压缩和展开张量
x = torch.arange(12)  
y = x.view(3, 4) 
z = x.view(-1, 2)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值