Of course.
So, assuming the volumes were 128x128x32, greyscale and batch was 2, instead of the batch being
[2,1,32,128,128] they would be [64,1,128,128].
I’m unsure if this sort of conversion even makes sense in any scenario, but I thought it could be an alternative assume there was no function to read and save the volumes as slices on the Dataset object.