728x90 dim-mismatch1 [PYTORCH] torch.cat()과 torch.stack()의 결정적 차이 1가지와 실무 해결 방법 7선 딥러닝 모델, 특히 파이토치(PyTorch)를 이용해 복잡한 네트워크를 설계하다 보면, 여러 소스에서 생성된 텐서(tensor)를 하나로 합쳐야 하는 상황을 끊임없이 마주하게 됩니다. 예를 들어, RNN이나 Transformer의 각 타임스텝에서 나온 hidden state들을 하나로 모으거나, 여러 이미지 배치를 합쳐 더 큰 배치를 만드는 경우입니다. 이때 우리는 torch.cat()과 torch.stack()이라는 두 가지 강력한 도구를 사용합니다. 하지만 많은 개발자가 이 두 함수의 동작 방식을 명확히 구분하지 못해 의도치 않은 형태(shape)의 텐서를 생성하고, 이는 곧 'dimension mismatch'라는 흔하지만 치명적인 런타임 에러로 이어집니다. 본 포스팅에서는 단순한 정의를 넘어, 이.. 2026. 4. 5. 이전 1 다음 728x90
