torch.unsqeeze
时间: 2023-10-15 07:30:01 浏览: 25
torch.unsqueeze函数是PyTorch中的一个函数,用于增加一个维度。它接受两个参数,第一个参数是要增加维度的张量,第二个参数是要添加的维度的索引。例如,如果有一个形状为(2,)的张量b,使用torch.unsqueeze(b, 0)将在索引0处添加一个维度,使其形状变为(1, 2)。同样,使用torch.unsqueeze(b, 1)将在索引1处添加一个维度,使其形状变为(2, 1)。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* [PyTorch的学习笔记01-基础中的基础](https://blog.csdn.net/huyaoyu/article/details/78526675)[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: 50%"]
- *3* [pytorch中的cat、stack、tranpose、permute、unsqeeze](https://blog.csdn.net/weixin_30782293/article/details/94844132)[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: 50%"]
[ .reference_list ]