unsqueeze(

unsqueeze()是PyTorch中的一个函数，用于在张量的指定维度上增加一个维度。

使用unsqueeze()函数时，可以传入两个参数，第一个参数是输入张量，第二个参数是要在哪个维度上增加一个维度。也可以直接在张量上调用unsqueeze()方法，并传入要增加维度的位置参数。

与unsqueeze()相对的是squeeze()函数，squeeze()函数用于压缩张量的维度。squeeze()函数可以去除张量中维度为1的维度。

举个例子，假设有一个维度为(3,)的张量a，如果我们想在这个张量的第二个维度上增加一个维度，可以使用unsqueeze(1)或a.unsqueeze(1)。这样，张量a的维度将变为(3, 1)。

另外，unsqueeze()函数还支持使用负数索引来指定要增加维度的位置，例如可以使用b.unsqueeze(-2)来在倒数第二个维度上增加一个维度。

综上所述，unsqueeze()函数主要用于在张量的指定维度上增加一个维度，而squeeze()函数则用于压缩张量的维度。<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中squeeze()和unsqueeze()函数介绍](https://download.csdn.net/download/weixin_38629976/13707836)[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* [【学习笔记】pytorch中squeeze()和unsqueeze()函数介绍](https://blog.csdn.net/flysky_jay/article/details/81607289)[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 ]