pytorch 增squeeze

在PyTorch中，squeeze()方法用于对Tensor对象进行维度压缩，即去除维数为1的维度。可以通过以下方式使用squeeze()方法：

import torch

x = torch.rand(3, 2, 1, 2, 1)
x = x.squeeze()  # 去除所有维数为1的维度
print(x.size())  # 输出：torch.Size(\[3, 2, 2\])

在上述代码中，我们首先创建了一个大小为3×2×1×2×1的Tensor对象x。然后使用squeeze()方法去除了所有维数为1的维度，最终得到了一个大小为3×2×2的Tensor对象。需要注意的是，squeeze()方法并不改变存储的数据，只是去除了"多余"的维度信息。
#### 引用[.reference_title]
- *1* *2* *3* [Pytorch squeeze()的简单用法](https://blog.csdn.net/qq_40714949/article/details/112770987)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]