ValueError: Target size (torch.Size([10, 1])) must be the same as input size (torch.Size([10, 2]))

Question

A binary classification problem with Batch Size = 10. Trying to use torch.nn.BCEWithLogitsLoss().

~\\Anaconda3\\envs\\notebook\\lib\\site-packages\\torch\\nn\\functional.