[PyTorch] cat() vs stack()
딥러닝 실습을 하다 보면 데이터를 합치는 작업이 꼭 필요합니다.PyTorch에서는 주로 torch.cat()과 torch.stack() 두 가지 함수를 사용하는데, 저도 매번 헷갈려서 이번에 한번 쭉 정리해보려 합니다.torch.cat() "이미 있는 축(axis) 방향으로 나란히 연결한다." cat은 같은 차원끼리 이어붙이는 함수입니다. 차원 수는 변하지 않습니다. input들의 shape이 모든 차원에서 일치해야 하며, 붙이려는 차원만 다를 수 있습니다.1. torch.cat((a, b), dim=0)a = torch.tensor([[1, 2], [3, 4]]) # shape: (2, 2)b = torch.tensor([[5, 6], [7, 8]]) # shape: (2, 2)# dim=0 (..
2025. 4. 29.