How to slice 3D tensor to 2D slices and construct Dataloader

[roaffix (Anton)]

Hi,

Have you checked this topic Train 3D Networks with Pytorch?