반응형 분류 전체보기733 [pytorch] unsqueeze 함수, 차원 늘리기 이번 글에서는 torch tensor의 차원을 늘리는 방법중에 하나인 unsqueeze함수에 대해 알아보겠다. 이 글을 보기 전에 차원의 숫자 1인 차원을 없애는 방법인 squeeze를 봐고 와도 좋다. ([pytorch] squeeze사용법: 텐서의 차원 제거하기)안봐도 된다. Unsqueeze 함수 사용법import torch tensor = torch.rand(10, 64, 128)tensor = tensor.unsqueeze(dim=2) #[10, 64, 128] -> [10,64,1,128]위와 같이 내가 차원을 증가시키고 싶은 차원에 해당하는 인덱스를 입력하면 차원이 증가하는 것을 확인할 수 있다. 2024. 9. 24. 이전 1 ··· 228 229 230 231 232 233 234 ··· 733 다음 반응형