Solved by albanD in post #7 
              
              
                Hi, 
I am afraid functions like .index_select() are not available at the moment and you would need them to get a mini batch from your dataset. 
You could potentially keep your array as a scipy one. Extract the minibatch from the scipy array (I expect this is possible but I don’t know). And then conv…