pytorch.chunk是挨着挨着进行分块的,其实这个可以借鉴Focus的思想,隔一个分一个块
代码:import numpy as np
import torch
data = torch.from_numpy(np.random.rand(1, 6, 3, 5))
print(str(data))
for i, data_i in enumerate(data.chunk(3, 1)):
print('#############################################################33')
print(str(data_i))
结果:



