Pytorch函数view、permute、squeeze、usqueeze（pytorch函数记不住）

Pytorch函数view、permute、squeeze、usqueeze（pytorch函数记不住）

1.view

在pytorch中view函数的作用为重构张量的维度，相当于numpy中resize()的功能，但是用法可能不太一样。

view有二种用法：

torch.view(参数a，参数b，。。。)torch.view(-1)或torch.view(参数a，-1)

总结一下：view的二种方法，一种就是当成resize一样转换数据shape，一种是如果是torch.view(参数a，-1)，则表示在参数b未知，参数a已知的情况下自动补齐列向量长度，在这个例子中a=3，x总共由9个元素，则b=9/3=3。以此类推更多维度的比如说

2.permute

3.squeeze

对数据的维度进行压缩使用方式： torch.squeeze(input, dim=None, out=None) 将输入张量形状中的1 去除并返回。 如果输入是形如(A×1×B×1×C×1×D)，那么输出形状就为： (A×B×C×D) 当给定dim时，那么挤压操作只在给定维度上。 例如，输入形状为: (A×1×B), squeeze(input, 0) 将会保持张量不变，只有用 squeeze(input, 1)，形状会变成 (A×B)。

4.usqueeze

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。