Solved by ptrblck in post #2 
              
              
                I’m not sure I understand the use case completely, but you could set batch_size=1 to get single samples in each batch (and remove the batch dimension if needed). 
Would this work for you? If not, could you explain what “removing batching” would mean?