이 블로그에서는 PyTorch에서 “torch.argmax()” 메서드를 사용하는 방법을 예시합니다.
PyTorch에서 'torch.argmax()' 메서드를 사용하는 방법은 무엇입니까?
'torch.argmax()' 메서드는 1D 또는 2D 텐서를 입력으로 사용하고 주어진 차원에 따른 최대값의 인덱스/인덱스를 포함하는 텐서를 반환합니다.
'torch.argmax()' 메서드의 구문은 다음과 같습니다.
토치. 인수최대 ( < 입력_텐서 > )
PyTorch에서 이 방법을 사용하려면 더 나은 이해를 위해 다음 예제를 살펴보세요.
예제 1: 1D Tensor에 'torch.argmax()' 메서드 사용
첫 번째 예에서는 1D 텐서를 생성하고 'torch.argmax()' 메서드를 사용합니다. 아래의 단계별 절차를 따르십시오.
1단계: PyTorch 라이브러리 가져오기
먼저 '를 가져옵니다. 토치 'torch.argmax()' 메소드를 사용하기 위한 라이브러리:
수입 토치2단계: 1D 텐서 생성
그런 다음 1D 텐서를 생성하고 해당 요소를 인쇄합니다. 여기서는 다음과 같은 '를 생성합니다. 텐스1 '를 사용하여 목록에서 텐서를 생성합니다. 토치.텐서() ' 기능:
텐스1 = 토치. 텐서 ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )
인쇄 ( 텐스1 )
그러면 아래와 같이 1D 텐서가 생성되었습니다.
3단계: 최대값 지수 찾기
이제 “ 토치.argmax() ” 함수를 사용하여 “에서 최대값의 인덱스/인덱스를 찾습니다. 텐스1 ' 텐서:
T1_ind = 토치. 인수최대 ( 텐스1 )4단계: 최대값 인덱스 인쇄
마지막으로 입력 텐서에 최대값의 인덱스를 표시합니다.
인쇄 ( '지수:' , T1_ind )아래 출력은 '의 최대값 인덱스를 보여줍니다. 텐스1 ” 텐서, 즉 4. 이는 텐서의 가장 높은 값이 4번째 인덱스에 있다는 것을 의미합니다. 9 ':
예제 2: 2D 텐서에 'torch.argmax()' 메서드 사용
두 번째 예에서는 2D 텐서를 생성하고 'torch.argmax()' 메서드를 사용합니다. 제공된 단계를 따르겠습니다.
1단계: PyTorch 라이브러리 가져오기
먼저 '를 가져옵니다. 토치 'torch.argmax()' 메소드를 사용하기 위한 라이브러리:
수입 토치2단계: 2D 텐서 생성
그런 다음 ' 토치.텐서() ” 함수를 사용하여 2D 텐서를 생성하고 해당 요소를 인쇄합니다. 여기서는 다음과 같은 '를 생성합니다. 텐스2 “2D 텐서:
텐스2 = 토치. 텐서 ( [ [ 4 , 1 , - 7 ] , [ 열 다섯 , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )인쇄 ( 텐스2 )
이는 아래와 같이 2D 텐서를 생성했습니다.
3단계: 최대값 지수 찾기
이제, “에서 최대값의 인덱스를 찾으세요. 텐스2 ”를 활용하여 텐서 토치.argmax() ' 기능:
T2_ind = 토치. 인수최대 ( 텐스2 )4단계: 최대값 인덱스 인쇄
마지막으로 입력 텐서에 최대값의 인덱스를 표시합니다.
인쇄 ( '지수:' , T2_ind )아래 출력에 따르면 ' 텐스2 ” 텐서는 “3”입니다. 이는 텐서의 가장 높은 값이 3번째 인덱스에 있다는 것을 의미합니다. 열 다섯 ':
5단계: 열을 따라 최대값의 인덱스 찾기
또한 사용자는 텐서의 각 열을 따라 최대값의 인덱스/인덱스를 찾을 수도 있습니다. 예를 들어, ' 어두움=0 ” 인수를 “torch.argmax()” 함수로 사용합니다. '의 열을 따라 최대값의 인덱스를 찾습니다. 텐스2 ” 텐서를 실행한 다음 해당 인덱스를 인쇄합니다.
col_index = 토치. 인수최대 ( 텐스2 , 어둑한 = 0 )인쇄 ( '열의 인덱스:' , col_index )
아래 출력은 텐서의 각 열에 따른 최대값의 인덱스를 보여줍니다.
6단계: 행을 따라 최대값의 인덱스 찾기
마찬가지로 사용자는 텐서의 각 행을 따라 최대값의 인덱스/인덱스를 찾을 수도 있습니다. 예를 들어, “ 어두움=1 '인수를 'torch.argmax()' 함수와 함께 사용하여 'Tens2' 텐서의 행을 따라 최대값의 인덱스를 찾은 다음 해당 인덱스를 인쇄합니다.
행_인덱스 = 토치. 인수최대 ( 텐스2 , 어둑한 = 1 )인쇄 ( '행의 인덱스:' , 행_인덱스 )
'Tens2' 텐서의 각 행에 있는 최대값 인덱스는 아래에서 볼 수 있습니다.
PyTorch에서 “torch.argmax()” 메서드를 사용하는 방법을 효율적으로 설명했습니다.
메모 : 여기에서 Google Colab Notebook에 액세스할 수 있습니다. 링크 .
결론
PyTorch에서 “torch.argmax()” 메서드를 사용하려면 먼저 “ 토치 ' 도서관. 그런 다음 원하는 1D 또는 2D 텐서를 생성하고 해당 요소를 봅니다. 다음으로 “ 토치.argmax() ” 텐서의 최대값에 대한 인덱스/인덱스를 찾고 계산하는 방법입니다. 또한 사용자는 '를 사용하여 텐서의 각 행이나 열에서 최대값의 인덱스를 찾을 수도 있습니다. 어둑한 ' 논쟁. 마지막으로 입력 텐서에 최대값의 인덱스를 표시합니다. 이 블로그에서는 PyTorch에서 'torch.argmax()' 메서드를 사용하는 방법을 예시했습니다.