目录
一、概述
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)
PyTorch中的view函数:张量形状变换与神经网络应用

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

被折叠的 条评论
为什么被折叠?



