[파이토치] torch.argmax 함수
torch.argmax torch.argmax(input) → LongTensor torch.argmax(input, dim, keepdim=False) → LongTensor 이 함수는 input tensor에 있는 모든 element들 중에서 가장 큰 값을 가지는 공간의 인덱스 번호를 반환하는 함수이다. a = torch.randn(4, 4) print(a) output = torch.argmax(a) print(output) #### output ##### ''' tensor([[-0.5014, -0.1785, 0.2534, 0.7167], [-0.7887, 1.0920, 0.5385, -1.1797], [-1.0129, 0.2337, 0.5757, 0.9139], [ 1.4672, -1.0605,..