Hi,



 Brando_Miranda:

Rows only exist in 2D tensors (matrices), at least to me


PyTorch modules such as Conv or Linear, only accept batched data, so if you have a single image you still have to create batches of size 1. So, not matter what type of data you are working with, you at least have 2D tensor for input and obviously, your model in the simplest case will generate a prob/logit for each sample in each batch where in case of a single instance in a batch, it would be [1, number_of_classes]. Now, taking max will give you value of max and index of max for each row which corresponds to each sample in each batch.
Bests