切块操作split and chunk
2022-06-20 本文已影响0人
菌子甚毒
https://blog.csdn.net/foneone/article/details/103875250
- torch.split(待操作tensor，切块后形状，dim)
- torch.chunk(待操作tensor, 切成多少块, dim)
t = torch.rand(10,3)
t1 = t.split(2,dim=0) # tuple (5,) t1[0]:torch.Size([2, 3])
t2 = t.chunk(2,dim=0) # tuple (2,) t2[0]:torch.Size([5, 3])
