torch.squeeze()

torch.squeeze()是一个PyTorch中的函数，用于从张量中删除尺寸为1的维度。它可以有两种用法：
1. torch.squeeze(input)：这种用法将张量input中的所有尺寸为1的维度都删除。例如，如果输入张量的形状为，则torch.squeeze(input)的输出形状将为。
2. torch.squeeze(input, dim)：这种用法将张量input中指定的尺寸为1的维度删除。参数dim是一个整数，表示要删除的维度的索引。例如，如果输入张量的形状为，则torch.squeeze(input, 3)的输出形状将为。中的代码示例展示了如何使用torch.squeeze()函数。其中，a是一个形状为的张量，b是将a中所有尺寸为1的维度删除后得到的张量，c是将a中索引为0的尺寸为1的维度删除后得到的张量，d是将a中索引为3的尺寸为1的维度删除后得到的张量。输出展示了每个张量的形状。从输出中可以看出，b的形状为，c的形状与a相同，d的形状为。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* [pyTorch.randn()、rand()、cat()、pow()、scatter_()、.squeeze() 、.unsqueeze()](https://download.csdn.net/download/weixin_38606656/14884199)[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^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 50%"]
- *2* *3* [torch.squeeze用法](https://blog.csdn.net/weixin_47156261/article/details/116600715)[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^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]