torch.cat()과 torch.stack()의 차이점
파이토치에서 텐서들을 서로 병합하는(붙이는) 2가지 함수 torch.cat()과 torch.stack()의 차이점에 대해 알아봅니다. torch.cat()은 주어진 차원을 기준으로 주어진 텐서들을 붙입니다(concatenate). torch.stack()은 새로운 차원으로 주어진 텐서들을 붙입니다. 따라서, (3, 4)의 크기(shape)를 갖는 2개의 텐서 A와 B를 붙이는 경우, torch.cat([A, B], dim=0)의 결과는 (6, 4)의 크기(shape)를 갖고, torch.stack([A, B], dim=0)의 결과는 (2, 3, 4)의 크기를 갖습니다. 예를 들어 설명하기 위해, 아래 두 개의 텐서 t1, t2를 예시로 선언해보겠습니다. t1 = torch.tensor([[1, 2], [..
