So for binary classification problem (Y in {0,1}), if I want to use nn.BCEWithLogitsLoss, then the output of my model should be raw scores for class=1 and have shape=(N) where N is the batch size, right?