Hi Mailcorahul,
Thank you very much for your kind response.Yes the labels tensor is like this [1, 0, 1, 0, 0] but the outputs tensor size is (N, 2) and Yes, this is a binary classification task…please below…
print(inputs.shape)
print(labels.shape)
print(outputs.shape)
print(labels)
print(outputs)
torch.Size([100, 4])
torch.Size([100])
torch.Size([100, 1])
tensor([1, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0,
0, 1, 0, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0,
0, 1, 0, 0, 0, 1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 0, 0, 0,
0, 0, 1, 0])
tensor([[  -0.6877],
[ -79.4583],
[  -0.6877],
[  -0.6877],
[  -0.6877],
[-231.3748],