torch-calc-function

• 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 반환