PyTorch中.unsqueeze()的使用方法和示例

pytorch 中.unsqueeze()
在 PyTorch 中，unsqueeze() 是一种常用的函数，用于在张量的特定维度上添加一个或多个维度。这对于调整张量的形状以匹配其他张量的形状或进行某些特定的操作非常有用。本文将深入探讨 PyTorch 中 unsqueeze() 的含义、使用方法和示例。
unsqueeze()函数的作用是在输入张量的所有维度上添加一个大小为1的维度。这个新的维度可以被放置在输入张量的任意位置，具体取决于参数dim。如果dim的值为正，则在对应的位置添加维度；如果dim的值为负，则在对应的位置从尾部添加维度。
unsqueeze()函数的输入参数如下：

1. input：这是需要执行unsqueeze()操作的张量。
2. dim：这是要添加的新维度的位置。可以是一个整数（正数或负数），也可以是一个元组，表示要在多个维度上添加新的维度。
   下面是一个例子，展示了如何使用unsqueeze()函数：

   import torch
   # 创建一个大小为(3, 4)的张量
   x = torch.rand(3, 4)
   print("原始张量 x：")
   print(x)
   # 在第0维（即第一个维度）上添加一个大小为1的维度
   y = x.unsqueeze(0)
   print("添加维度后的张量 y：")
   print(y)
   # 在第1维（即第二个维度）上添加一个大小为1的维度
   z = x.unsqueeze(1)
   print("添加维度后的张量 z：")
   print(z)

   输出结果如下：
   ```css
   原始张量 x： 0.5480 0.2247 0.5418 0.8739 0.0461 0.5756 0.9037 0.1434 0.1528 0.7917 0.3687 0.6209
   [torch.FloatTensor of size (3, 4)]
   添加维度后的张量 y： 0.5480 0.2247 0.5418 0.8739 0.0461 0.5756 0.9037 0.1434 0.1528 0.7917 0.3687 0.6209 [[torch.FloatTensor of size (4,)], [torch.FloatTensor of size (4,)], [torch.FloatTensor of size (4,)], [torch.FloatTensor of size (4,)]]] [[torch.FloatTensor of size (4,)], [torch.FloatTensor of size (4,)], [torch.FloatTensor of size (4,)], [torch.FloatTensor of size (4,)]]] 添加维度后的张量 z： 0.5480 0.2247 0.5418 0.8739 0.5480 0.2247 0.5418 0.8739 0. [[[torch.FloatTensor of size (1,)], [torch.FloatTensor of size (1,)], [torch.FloatTensor of size (1,)], [torch.FloatTensor of size (1,)]]]](https://pytorch.org/docs/stable/generated/torch.FloatTensor%20of%20size%20%281,%29%E3%80%82%E3%80%82%E