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

먼저 torch.squeeze()를 보면 이 함 수 는 주로 데이터 의 차원 을 압축 하고 차원 이 1 인 차원 을 제거한다.예 를 들 어 한 줄 이나 한 열 같은 것 을 제거한다.한 줄 의 세 열(1,3)의 수 는 첫 번 째 차원 이 1 인 차원 을 제거 한 후에(3)줄 이 된다.squeeze(a)는 a 의 모든 차원 을 1 로 삭제 하 는 것 이다.1 이 아 닌 차원 은 영향 이 없다.a...

pytorchsqueeze()unsqueeze()