4. Как получить предсказания? Пусть logits.shape == (360, 10). Как получить 360 номеров цифр?
logits.shape == (360, 10)