tensor.unsqueeze()

tensor.unsqueeze()是PyTorch中用于改变tensor形状的函数之一。它的作用是在指定的维度上增加一个维度。

举个例子来说明它的使用：假设我们有一个形状为(3, 2)的tensor，我们可以使用unsqueeze()函数在指定维度上增加一个维度。比如，在维度0上增加一个维度，我们可以这样写：tensor.unsqueeze(dim=0)。这样会使得原本形状为(3, 2)的tensor变成形状为(1, 3, 2)的tensor。

总结来说，tensor.unsqueeze(dim)可以在指定的维度上增加一个维度，并返回新的tensor。默认情况下，dim为None，表示在所有维度上都增加一个维度。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [tensor.squeeze函数和tensor.unsqueeze函数的使用](https://blog.csdn.net/qq_41661809/article/details/129332303)[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_2"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]