Doing torch.max, Error: Dimension out of range (expected to be in range of [-1, 0], but got 1)

Question

Below is my error message. I\'m confused on how to fix it.

---------------------------------------------------------------------------
IndexError