아래 내용은 pytorch 2.1.0 버전으로 작성됨.

0. 바로 사용하기

torch.stack((tensor1, tensor2, tensor3), dim=0)

• (tensor1, tensor2, tensor3): 새로운 차원에 쌓아올릴 텐서들
• dim=0: 쌓아올릴 차원

---

1. 기본형

torch.stack(tensors, dim=0, *, out=None) → Tensor

---

2. 기능

• 입력 받은 tensor들을 새로운 차원에 샇아 올려 만든 새로운 tensor를 반환함

---

3. 파라미터

tensors

• tensors: 쌓아 올릴 tensor
  • 샇아올릴 텐서들은 반드시 동일한 크기를 가져야 함

• (N, K) , (N, K) tensor를 stack하여 (2, N, K) tensor가 만들어짐
• ref (https://velog.io/@nochesita/PyTorch-torch.cat과-torch.stack의-차이)

dim

• dim = integer: tensor들을 쌓아올릴 차원

• (3, 3) tensor 2개를 dim=0으로 stack한 결과.
• 0차원 자리에 새로운 차원이 생성됨.

• dim=1로 설정할 경우 1차원 자리에 새로운 차원이 생성됨.

---

ref

• https://pytorch.org/docs/2.1/generated/torch.stack.html#torch.stack
• https://velog.io/@nochesita/PyTorch-torch.cat과-torch.stack의-차이