import torch
import numpy as np
维度变换
a = torch.randint(1,10,[4,1,3,6]) #4张照片,1个通道(即灰度图片),长是3个pixel,宽是6个pixel
a.reshape(-1,3,6) #reshape只要保证numel相同就可以了
tensor([[[6, 1, 8, 2, 3, 8],
[1, 7, 6, 9, 2, 7],
[1, 6, 2, 5, 7, 6]],
[[9, 4, 3, 9, 4, 2],
[4, 6, 8, 7, 7, 1],
[1, 3, 9, 7, 1, 8]],
[[6, 4, 5, 3, 6, 4],
[7, 9, 7, 1, 2, 1],
[1, 5, 7, 6, 5, 4]],
[[5, 6, 4, 2, 1, 6],
[9, 5, 4, 9, 3, 8],
[9, 1, 2, 4, 5, 8]]])
- squeeze and unsqueeze
a = torch.randint(1,10,[3,4])
a
tensor([[6, 6, 9, 7],
[6, 5, 1, 5],
[6, 3, 8, 9]])
a.unsqueeze(0) #unsqueeze括号里面填的是维数,取值范围是【