随笔记录——torch.squeeze/torch.unsqueeze

使用squeeze()移除输入的tensor中，大小为1的维度

import torch
x = torch.zeros(2, 1, 2, 1, 2)

y = torch.squeeze(x)
y.shape
torch.Size([2, 2, 2])

使用squeeze()移除输入的tensor中，指定维度的大小为1的维度

y = torch.squeeze(x, 0)
y.shape
torch.Size([2, 1, 2, 1, 2])

y = torch.squeeze(x, 1)
y.shape
torch.Size([2, 2, 1, 2])

y = torch.squeeze(x, 2)
y.shape
torch.Size([2, 1, 2, 1, 2])

y = torch.squeeze(x, dim=3)
y.shape
torch.Size([2, 1, 2, 2])

如果指定了dim，则如果对应dim的维的大小为1， 则指定维度被删除。

在指定位置插入一个大小为1的维度

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

torch.unsqueeze(x, 0).shape
torch.Size([1, 4])

torch.unsqueeze(x, 1).shape
torch.Size([4, 1])

暂时完结。。。