(Pdb) out.data.max(1)[1].cpu().numpy().max()
18295877782208576
(Pdb) out.data.max(dim=1)[1].cpu().numpy().max()
22518346029203522
(Pdb) out.data.max(dim=0)[1].cpu().numpy().max()
22518346030252096
(Pdb) out.data.max(dim=2)[1].cpu().numpy().max()
799
(Pdb) out.data.max(dim=3)[1].cpu().numpy().max()
799
(Pdb) out.data.max(dim=1)[1].cpu().numpy().max()
18014746402881610
(Pdb) out.data.max(dim=0)[1].cpu().numpy().max()
22518346030252096
(Pdb) out.size()
torch.Size([1, 35, 800, 800])

I guess while calculating out.data.max(dim=1)[1].cpu().numpy().max(), the output should be less than 35. any one could explain this ?