pytorch 에서 squeeze()와 unsqueze()함수 소개

squeeze 의 용법 은 주로 데이터 의 차원 을 압축 하거나 압축 을 푸 는 것 이다.
먼저 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()내용 은 우리 의 이전 글 을 검색 하거나 아래 의 관련 글 을 계속 찾 아 보 세 요.앞으로 많이 응원 해 주세요!

좋은 웹페이지 즐겨찾기