title: BCEWithLogitsLoss: Trying to get binary output for predicted label as a tensor, confused with output layer
body: <p>Each element of my dataset has a multi-label tensor like <code>[1, 0, 0, 1]</code> with varying combinations of <code>1's</code> and <code>0's</code>. In this scenario, since I have 4 tensors, I have the output layer of my neural network to be 4. In doing so with BCEWithLogitsLoss, I obtain an output tensor like <code>[3, 2, 0, 0]</code> when I call model(inputs) which is in the range of (0, 3) as I specified with the output layer to have 4 output neurons. This does not match the format of what the target is expected to be, although when I change the number of output neurons to 2, I get a shape mismatch error. What needs to be done to fix this?</p>