Yes, we support that. You can find modules like Conv3d and pooling for volumes. I don’t understand the question about generating batches, you need to write your own Dataset 100 that will load and return 4D elements from __getitem__. Then use a DataLoader 48 to load the data (you can give it a batch_size argument, and it will batch the 4D elements into a 5D tensor for you).