Just checked and you are right.
Sorry I didn’t link the link before:

  
      github.com/pytorch/pytorch
  
  
    



Issue: torch.argmin behaves differently on CPU and GPU 6


	
	opened by carefree0910
	on 2018-11-17
	
	
	closed by soumith
	on 2018-11-17
	


🐛 Bug
torch.argmin behaves differently on cpu (which is correct) and gpu (which is wrong)
I'm not sure whether this 'bug' (or 'feature')...




  
  
    
    
  
  


But according to that answer (which in current pytorch versions is no longer valid) CPU version of torch.max and torch.min always returned the correct argmax, which is not true anymore.