torch.unsqueeze()

torch.unsqueeze()函数是一个PyTorch中的函数，用于在指定维度上增加一个维度。它的作用是将一个张量的维度增加到指定的尺寸。

该函数的使用格式为torch.unsqueeze(input, dim)，其中input是要进行维度扩充的张量，dim是要插入的维度的索引。使用时，也可以直接写成input.unsqueeze(dim)。

该函数在多种情况下都有用途，例如在进行卷积操作时，需要将输入的维度扩充到适当的大小。通过增加维度，可以满足不同操作的输入要求。

举个例子，如果我们有一个形状为(3, 5)的张量，我们可以使用unsqueeze函数在维度1上增加一个维度，从而得到一个形状为(3, 1, 5)的新张量。这样做可以方便在某些操作中使用。

总结来说，torch.unsqueeze()函数可以在指定维度上增加一个维度，使得张量的形状适应不同的操作需求。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* [torch.unsqueeze()函数理解](https://blog.csdn.net/weixin_53142585/article/details/128997156)[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^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"]
- *2* [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^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"]
- *3* [torch.unsqueeze() 和 torch.squeeze()](https://blog.csdn.net/qq_36387683/article/details/108093536)[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^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"]
[ .reference_list ]