pytorch 에서 squeeze()와 unsqueze()함수 소개
1787 단어 pytorchsqueeze()unsqueeze()
먼저 torch.squeeze()를 보면 이 함 수 는 주로 데이터 의 차원 을 압축 하고 차원 이 1 인 차원 을 제거한다.예 를 들 어 한 줄 이나 한 열 같은 것 을 제거한다.한 줄 의 세 열(1,3)의 수 는 첫 번 째 차원 이 1 인 차원 을 제거 한 후에(3)줄 이 된다.squeeze(a)는 a 의 모든 차원 을 1 로 삭제 하 는 것 이다.1 이 아 닌 차원 은 영향 이 없다.a.squeeze(N)는 a 에서 지정 한 차원 을 1 로 제거 하 는 것 이다.또 다른 형식 은 b=torch.squeeze(a,N)a 에서 지정 한 위 수 를 1 로 하 는 차원 을 제거 하 는 것 이다.
torch.unsqueeze()를 보면 이 함 수 는 주로 데이터 차원 을 확장 합 니 다.지 정 된 위치 에 위 수 를 1 로 하 는 차원 을 추가 합 니 다.예 를 들 어 원래 세 줄 의 데이터(3)가 있 었 는데 0 의 위치 에 1 차원 을 더 하면 한 줄 의 3 열(1,3)이 됩 니 다.a.squeeze(N)는 a 에서 지정 한 위치 N 에 1 의 차원 을 더 하 는 것 이다.또 다른 형식 은 b=torch.squeeze(a,N)a 는 a 에서 지정 한 위치 N 에 1 차원 을 더 하 는 것 이다.
1.unsqueeze()함수
1.우선 a 를 초기 화 합 니 다.
a 의 차원 은(2,3)임 을 알 수 있다.
2.2 차원 에서 하나의 차원 을 추가 하여 그 차원 을(2,1,3)로 바꾼다.
a 의 차원 이(2,1,3)로 바 뀌 었 음 을 알 수 있 습 니 다.마찬가지 로 마지막 두 번 째 차원 에서 하나의 차원 을 추가 해 야 한다 면 b.unsqueeze(-2)를 사용 하 십시오.
2.squeeze()함수 소개
1.우선 하나의 차원(1,2,3)의 tensor(장 량)를 얻는다.
그림 에서 알 수 있 듯 이 c 의 차원 은(1,2,3)이다.
2.squeeze()함 수 를 사용 하여 1 차원 을 제거 합 니 다.
이 를 통 해 알 수 있 듯 이 차원 이(2,3)로 바 뀌 었 다.
3.그리고
차원 이 변 하지 않 고(1,2,3)인 것 을 알 수 있 는데 이것 은 차원 이 1 일 때 만 없어 지기 때문이다.
pytorch 의 squeeze()와 unsqueeze()함수 에 대한 자세 한 설명 은 여기까지 입 니 다.더 많은 pytorch 의 squeeze()와 unsqueeze()내용 은 우리 의 이전 글 을 검색 하거나 아래 의 관련 글 을 계속 찾 아 보 세 요.앞으로 많이 응원 해 주세요!
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
정확도에서 스케일링의 영향데이터셋 스케일링은 데이터 전처리의 주요 단계 중 하나이며, 데이터 변수의 범위를 줄이기 위해 수행됩니다. 이미지와 관련하여 가능한 최소-최대 값 범위는 항상 0-255이며, 이는 255가 최대값임을 의미합니다. 따...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.