Pytorch

1.torch.argmax() & torch.max()

post-thumbnail

Pytorch docs : torch.argmax()위의 결과를 보면 알 수 있듯이, 1행 → 4행의 순서로 큰 숫자를 파악하여 tensor를 출력한다.dim = 0의 파라미터를 추가할 경우, 결과는 위와 같다. 열을 기준으로 max 값이 있는 index를 출력한다.이

2021년 8월 26일