PyTorch 텐서의 크기를 조정하려면 .view()를 사용합니다. 방법. 텐서의 차원을 늘리거나 줄일 수 있지만 크기 조정 전후에 텐서의 총 요소 수가 일치해야 합니다.
단계
-
필요한 라이브러리를 가져옵니다. 다음 모든 Python 예제에서 필수 Python 라이브러리는 torch입니다. . 이미 설치했는지 확인하십시오.
-
PyTorch 텐서를 만들고 인쇄합니다.
-
.view()를 사용하여 위에서 만든 텐서의 크기를 조정합니다. 변수에 값을 할당합니다. .view() 원래 텐서의 크기를 조정하지 않습니다. 이름에서 알 수 있듯이 새 크기의 보기만 제공합니다.
-
마지막으로 크기 조정 후 텐서를 인쇄합니다.
예시 1
# PyTorch에서 텐서 크기를 조정하는 Python 프로그램# 라이브러리 가져오기 토치# 텐서 생성T =torch.Tensor([1, 2, 3, 4, 5, 6])print(T)# T를 2x3x로 크기 조정 =T.view(2,3)print("크기 조정 후 텐서:\n",x)# T를 2x3x로 크기 조정하는 다른 방법 =T.view(-1,3)print("크기 조정 후 텐서:\n", x)# 다른 방법으로 T를 2x3x로 크기 조정 =T.view(2,-1)print("크기 조정 후 텐서:\n",x) 출력
위의 Python 3 코드를 실행하면 다음 출력이 생성됩니다.
텐서([1., 2., 3., 4., 5., 6.])크기 조정 후 텐서:tensor([[1., 2., 3.], [4., 5., 6.]])크기 조정 후 텐서:텐서([[1., 2., 3.], [4., 5., 6.]])크기 조정 후 텐서:텐서([[1., 2., 3] .], [4., 5., 6.]])
예시 2
# 라이브러리 가져오기import torch# 텐서 모양 만들기 4x3T =torch.Tensor([[1,2,3],[2,1,3],[2,3,5],[5,6,4 ]])print(T)# T를 3x4x로 크기 조정 =T.view(-1,4)print("크기 조정 후 텐서:\n",x)# T를 3x4x로 크기 조정하는 다른 방법 =T.view(3, -1)print("크기 조정 후 텐서:\n",x)# T를 2x6x로 크기 조정 =T.view(2,-1)print("크기 조정 후 텐서:\n",x) 출력
위의 Python 3 코드를 실행하면 다음 출력이 생성됩니다.
텐서([[1., 2., 3.], [2., 1., 3.], [2., 3., 5.], [5., 6., 4.]] )크기 조정 후 텐서:텐서([[1., 2., 3., 2.], [1., 3., 2., 3.], [5., 5., 6., 4.]] )크기 조정 후 텐서:텐서([[1., 2., 3., 2.], [1., 3., 2., 3.], [5., 5., 6., 4.]] )크기 조정 후 텐서:tensor([[1., 2., 3., 2., 1., 3.], [2., 3., 5., 5., 6., 4.]])