So if I have 5 output classes and 3 test instances and for example I will take the below output:
tensor([[ 0.4657, -0.7640, -1.4268, -0.5012,  1.2167],
[-0.4578, -0.5621, -0.5652, -0.4056,  0.2509],
[-0.5617,  0.8141, -0.1722, -0.1264,  0.2285]]
this means that for the first instance the right class is the 5th? for the second also the 5h and of the last instance the 2nd?  (the biggest ones) ??