刪除tensor中不是1的維度

squeeze只能刪除是1的維度，因此對不是1的維度先切片為1，然後再刪除

input_torch=torch.randn([1,5,1,2])
print(input_torch[:, :, :1, :1].squeeze(dim=(2, 3)).shape) #先切片為[1,5,1,1]，再刪除後兩維