深入浅出Pytorch函数——torch.unsqueeze

分类目录：《深入浅出Pytorch函数》总目录
相关文章：
· 深入浅出Pytorch函数——torch.squeeze
· 深入浅出Pytorch函数——torch.unsqueeze

---

返回一个新的张量，且在指定位置新插入了一个大小为1的维度，返回的张量与该张量共享相同的底层数据。

语法

torch.unsqueeze(input, dim) → Tensor

参数

• input：[Tensor] 输入张量
• dim：[int] 插入维度的位置索引，范围是$[\text{-input.dim()}-1,\text{input.dim()}+1]$，默认值为$\text{input.dim()}+1$

实例

输入：

x = torch.tensor([1, 2, 3, 4])
torch.unsqueeze(x, 0)

输出：

tensor([[ 1,  2,  3,  4]])

输入：

x = torch.tensor([1, 2, 3, 4])
torch.unsqueeze(x, 1)

输出：

tensor([[1],
        [2],
        [3],
        [4]])