- toc {:toc}
torch.max(input, dim, keepdim=False, *, out=None)
- input tensor의 모든 요소중 최댓값을 출력한다.
- parameter
-
dim - 값에 따라 각 차원에서 최댓값을 뽑아낸다.
ex) 2차원일 때, 0이면 열에서, 1이면 행에서 최댓값 산출
-
keepdim - 입력 차원을 유지할지 선택한다. default = False
-
- return
- input만 입력 → 최댓값만 반환
- input, dim,… 입력 → 결과값, 값을 갖는 index 반환