torch.unsqueeze()和torch.squeeze()函数如何使用

# torch.unsqueeze()和torch.squeeze()函数如何使用

在PyTorch中，`torch.unsqueeze()`和`torch.squeeze()`是常用的张量维度操作函数，用于调整张量的形状。

## 1. torch.unsqueeze()
`torch.unsqueeze(input, dim)`函数用于在指定维度`dim`上增加一个大小为1的维度。例如：
```python
import torch
x = torch.tensor([1, 2, 3])  # 形状 [3]
y = torch.unsqueeze(x, 0)    # 形状变为 [1, 3]

这在需要扩展维度进行广播或模型输入时非常有用。

2. torch.squeeze()

torch.squeeze(input, dim=None)函数用于移除所有大小为1的维度（若指定dim则仅移除该维度）。例如：

z = torch.tensor([[[1], [2]]])  # 形状 [1, 2, 1]
w = torch.squeeze(z)            # 形状变为 [2]

常用于去除不必要的单维度，简化张量结构。

总结

• unsqueeze()：增加维度，用于维度扩展
• squeeze()：压缩维度，用于简化形状 两者配合使用可以灵活调整张量维度以适应不同计算需求。

”`