I have been trying to implement the loss over a mini batch of 32. I have a total dataset of 1024 and i split the data in 32 different mini batches. I am trying to subtract e