torch.unsqueeze() 函数
torch.unsqueeze(input, dim) → Tensor
unsqueeze()函数起升维的作用，对输入的既定位置（dim）升一维。
dim范围在:[-input.dim() - 1, input.dim() + 1)之间，比如输入input是一维，dim可以是-2，-1，0，1，而负dim相当于 dim = dim + input.dim() + 1，也就是dim实际上就只有0,1。
则dim=0时数据为行方向升维，dim=1时为列方向升维。
>>> x = torch.tensor([1, 2, 3, 4]) #torch.Size([4])
>>> torch.unsqueeze(x, 0)
tensor([[ 1, 2, 3, 4]]) #torch.Size([1, 4])
>>> torch.unsqueeze(x, 1)
tensor([[ 1],[ 2],[ 3],[ 4]]) #torch.Size([4, 1])
torch.squeeze(input, dim=None) → Tensor
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！