I have a batch of shape (20,28) and the output produced by the model has the same shape as I am not using sigmoid in the last layer.  I am using BCEWithLogitsLoss as a criterion. How can I pass the y_pred and y_actual to this criterion. I mean in which shape should I modify the shape of the target and predicted value?