How to predict only one test sample in pytorch model?

can you explain why:

torch.max(output.data, 1)

works?

I was reading the docs (https://pytorch.org/docs/stable/generated/torch.max.html) and it’s still not entirely clear to me. This is the sentence that doesn’t make sense to me:

Returns a namedtuple (values, indices) where values is the maximum value of each row of the input tensor in the given dimension dim.

I don’t understand what the max value of a row mean given a specific slice of a tensor. Rows only exist in 2D tensors (matrices), at least to me. Can you clarify?