i mean, for each batch the input of the loss function is a list of all predictions and labels in the current batch, and the loss is built for input of only one prediction and label
so how it should be implement?
and another thing - how the backward() of costume function should be implemented?