Screenshot (162)1920×1080 229 KB
size of the labels is: torch.size([50,10])
How to convert it to single dimension. Weather the batch size(50) is necessary or not.
Let me know the code for it plz.
https://www.kaggle.com/basavarajrp4444/kernel755d8a6dc5/edit/run/36854797
Thank you.