深度学习(14)--x.view()详解

在torch中,常用view()函数来改变tensor的形状

查询官方文档:

torch.Tensor.view --- PyTorch 2.2 documentationhttps://pytorch.org/docs/stable/generated/torch.Tensor.view.html#torch.Tensor.view示例

1.创建一个4x4的二维数组进行测试

python 复制代码
x = torch.randn(4, 4)
print(x)
print(x.size())

(1).将二维数组变换为一维数组

python 复制代码
y = x.view(16)
print(y)
print(y.size())

(2).将二维数组变换为其他形式的二维数组

python 复制代码
z = x.view(2, 8)
print(z)
print(z.size())

(3).可以将其中一个参数设置为-1,view()会根据已设置的维度自动推断出另外一个维度的大小

复制代码
# the size -1 is inferred from other dimensions
python 复制代码
yy = x.view(-1, 8)
print(yy)
print(yy.size())

zz = x.view(8, -1)
print(zz)
print(zz.size())

可以看到分别得到了2x8的yy和8x2的zz,符合实际的情况。

2.创建一个1x2x3x4的四维矩阵进行测试

python 复制代码
x = torch.rand(1, 2, 3, 4)
print(x)
print(x.size())

(1).将四维数组变换为一维数组

python 复制代码
y = x.view(-1)
print(y)
print(y.size())

(2).将四维数组变换为二维数组

python 复制代码
z = x.view(2,-1)
print(z)
print(z.size())

(3).将四维数组变换为三维数组

python 复制代码
a = x.view(2, -1, 4)
print(a)
print(a.size())

(4).将四维数组转换为其他形式的四维数组

python 复制代码
b = x.view(1, 3, 2, 4)
print(b)
print(b.size())

值得注意的是view()函数并不改变tensor数据在内存中的层次

利用tranpose函数进行验证,transpose函数可以交换数据指定的维度:

python 复制代码
c = x.transpose(1, 2)
print(c)
print(c.size())

transpose(1,2)将第二个维度和第三个维度互换(四维对应的索引是0,1,2,3)

利用equal()函数判断b和c是否相同:

python 复制代码
print("b和c是否相等:")
print(torch.equal(b, c))

由如上结果可知,view()函数并不改变数据在内存中的层次。

相关推荐
西柚小萌新3 分钟前
【从零开始的大模型原理与实践教程】--第一章:NLP基础概念
人工智能·自然语言处理
程序员奈斯6 分钟前
Python深度学习:NumPy数组库
python·深度学习·numpy
嘀咕博客9 分钟前
SafeEar:浙大和清华联合推出的AI音频伪造检测框架,错误率低至2.02%
人工智能·音视频·ai工具
Hello123网站9 分钟前
FinChat-金融领域的ChatGPT
人工智能·chatgpt·金融·ai工具
嘀咕博客14 分钟前
PixVerse -免费在线AI视频生成工具
人工智能·音视频·ai工具
CoovallyAIHub16 分钟前
CostFilter-AD:用“匹配代价过滤”刷新工业质检异常检测新高度! (附论文和源码)
深度学习·算法·计算机视觉
CoovallyAIHub24 分钟前
CVPR 2025 | 频率动态卷积(FDConv):以固定参数预算实现频率域自适应,显著提升视觉任务性能
深度学习·算法·计算机视觉
mit6.82427 分钟前
[rStar] 解决方案节点 | `BaseNode` | `MCTSNode`
人工智能·python·算法
普蓝机器人35 分钟前
AutoTrack-IR-DR200底盘仿真详解:为教育领域打造的高效机器人学习实验平台
人工智能·学习·计算机视觉·机器人·移动机器人·三维仿真导航
赴3351 小时前
opencv 银行卡号识别案例
人工智能·opencv·计算机视觉·银行卡号识别