Python——torch.unsqueeze

unsqueeze在指定位置處插入尺寸為1的維度。

注意：返回的張量與此張量共享相同的基礎數據。

 

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

 

import torch

torch中的squeeze與unsqueeze作用是去除/添加維度為1的行

例如，a=torch.randn(2,3)

那么b=a.unsqueeze(0)，b為（1,2,3）的矩陣

類似的，b=a.unsqueeze(1)，b為（2,1,3）的矩陣

b=a.unsqueeze(2)，b為（2,3,1）的矩陣