- 博客(1)
- 收藏
- 关注
原创 torch.chunk与torch.split的区别
torch.chunk(tensor, chunks, dim=0)将tensor 拆分成相应的组块, 最后一块会小一些如果不能整除的话。a的size是(2,4),那么torch.chunk(a,2,0), 0表示以第一个维度,分成两块,第一个维度就是指(2,4)里面的2torch.split(tensor, split_size_or_sections, dim=0)将tensor 拆分...
2020-04-28 14:42:12
1496
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人