이 기사에서는 PyTorch에서 Tensor에 차원을 추가하는 방법을 예시합니다.
PyTorch의 특정 텐서에 차원을 추가하는 방법은 무엇입니까?
사용자는 PyTorch의 1D 텐서 또는 2D 텐서와 같은 모든 Tensor에 차원을 추가할 수 있습니다. 특정 위치의 텐서에 새 차원을 추가하려면 더 나은 이해를 위해 다음 예를 확인하십시오.
예제 1: PyTorch의 1D Tensor에 차원 추가
이 예에서는 1D 텐서를 생성하고 특정 위치에 차원을 추가합니다. 실제 데모를 보려면 아래 나열된 단계를 따르십시오.
1단계: 라이브러리 가져오기
먼저 토치 라이브러리를 가져옵니다.
수입 토치
2단계: 1D 텐서 생성
그런 다음 1차원 텐서를 만듭니다. 예를 들어, 다음과 같은 텐서를 생성하여 ' 엑스 ” 변수:
엑스 = 토치. 텐서 ( [ 5 , 삼 , 8 , 2 ] )
3단계: 텐서 형태 보기
다음으로 새로 생성된 텐서 모양을 표시하여 해당 크기를 확인합니다.
아래 출력은 텐서가 1차원임을 나타냅니다.
4단계: 1D 텐서에 차원 추가
이제 “ torch.unsqueeze(입력, 희미함) ” 함수를 사용하여 특정 위치의 1D 텐서에 차원을 추가합니다. 예를 들어, 0 인덱스의 텐서에 차원을 추가합니다.
여기,
- “ new_tens ”는 추가된 차원을 포함하는 변수입니다.
- “ 엑스 ”는 입력 텐서입니다.
- “ 어두움=0 ”는 0 인덱스에 차원을 추가하는 데 사용됩니다.
5단계: 출력 확인
마지막으로 텐서에 새 차원이 추가되었는지 확인하세요.
아래 출력에서는 0 인덱스의 1D 텐서에 새 차원이 추가된 것을 확인할 수 있습니다.
또한 사용자는 다른 위치에 치수를 추가할 수도 있습니다. 여기에서는 첫 번째 인덱스에 차원을 추가했습니다.
예시 2: PyTorch의 2D 텐서에 차원 추가
여기서는 2D 텐서를 생성/만들고 특정 위치에 차원을 추가하겠습니다. 실제 구현을 위해 제공된 단계를 시도해 보세요.
1단계: Torch 라이브러리 가져오기
먼저 토치 라이브러리를 가져옵니다.
2단계: 2D 텐서 생성
그런 다음 2차원 텐서를 만듭니다. 예를 들어, 다음과 같은 텐서를 생성하여 ' 엑스 ” 변수:
3단계: 텐서 형태 보기
그런 다음 새로 생성된 텐서 모양을 표시하여 해당 크기를 확인합니다.
아래 출력에 따르면 이 텐서는 2차원입니다.
4단계: 2D 텐서에 차원 추가
이제 다음을 사용하여 특정 위치의 2D 텐서에 차원을 추가합니다. 'torch.unsqueeze(입력, 희미함)' 기능. 예를 들어, 0 인덱스의 텐서에 차원을 추가합니다.
5단계: 출력 확인
마지막으로, 새 차원이 2D 텐서에 추가되었는지 여부를 확인하세요.
아래 출력은 새 차원이 0 인덱스의 2D 텐서에 성공적으로 추가되었음을 나타냅니다.
메모 : 여기에서 Google Colab Notebook에 액세스할 수 있습니다. 링크 .
우리는 다양한 예제를 사용하여 PyTorch에서 텐서에 차원을 추가하는 방법을 효율적으로 설명했습니다.
결론
PyTorch의 텐서에 차원을 추가하려면 먼저 PyTorch 라이브러리를 가져옵니다. 그런 다음 1D 또는 2D 텐서를 생성하고 해당 크기를 확인합니다. 그런 다음 '를 사용하여 특정 위치의 텐서에 차원을 추가합니다. torch.unsqueeze(입력, 희미함) ' 기능. 사용자는 입력 텐서와 원하는 인덱스 위치를 이 함수에 매개변수로 전달해야 합니다. 이 기사에서는 PyTorch의 텐서에 차원을 추가하는 방법을 예시했습니다.