Как красиво вывести результат модели в pytorch?

Question

[andrew46]

Не так давно начал осваивать 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])

Вопрос, как мне вывести к какому классу модель относит цифру с картинки? Или какое-то значение, которое бы точно дало понять, что модель определила цифру верно.
Заранее извиняюсь за нубский вопрос)