chunk分块 python

import torch
a = torch.randn(2,3)
b = a.unsqueeze(2).unsqueeze(3)
c,d = b.chunk(2,1) #chunk(a,b),a表示分成的块数，b=0沿横向分割，b=1沿纵向分割
print(a)
print(',,,,,,,,,,,,,')
print(b.shape) #[2,3,1,1]
print(b)
print('.................')
print(c.shape) #[2,2,1,1]
print(c)
print('////////////////////')
print(d.shape) #[2,1,1,1]
print(d)

import torch
a = torch.randn(2,3)
b = a.unsqueeze(2).unsqueeze(3)
c,d = b.chunk(2,0)

print(b.shape) #[2, 3, 1, 1]
print('.................')
print(c.shape) #[1, 3, 1, 1]
print('////////////////////')
print(d.shape) #[1, 3, 1, 1]