I am working with a dataset of videos, I have done a preprocessing to split each video into a set of frames. I am implementing a network that uses a 3dconvolution where I have to pass chunks of videos to be processed by the network.
I know that a 3dConv layer takes a tensor of 5D (N, C, D, H,W). How do I modify the data  loader so that I get the data in the appropriate shape?