nmvr:

Or, in case it is not possible, is there a way to change the batches so that instead of batching out volumes, it batches out slices?
e.g if batch size is 2, instead of 2 volumes it would send out all slices from the 2 volumes.


Could you explain what the difference would be in these two cases, i.e. how the batch shape would look like?