Creating 3D Dataset/DataLoader with patches

26

unfold should work. Have a look at this post for an example.