pytorch: fast way to get the indices of the largest element of a matrix

Question

If I have a tensor (n*n matrix) [[1,2,3],[3,4,5],[5,6,7]], how can i get the indices of the largest element ([2,2]) using pytorch efficiently?

I tried to do t.max(mat