Не так давно начал осваивать PyTorch и появился такой вопрос. Есть натренированная модель для распознавания цифр с картинки, загружаю в нее свою картинку с цифрой таким образом:
img = Image.open('./data/9.jpg')
img_t = transform(img)
batch_t = torch.unsqueeze(img_t, 0)
out = model(batch_t)
print(out. shape)
Вывод: torch.Size([1, 10])
Вопрос, как мне вывести к какому классу модель относит цифру с картинки? Или какое-то значение, которое бы точно дало понять, что модель определила цифру верно.
Заранее извиняюсь за нубский вопрос)
