Tensor 변환과 결합 - cat(), view(), squeeze(), unsqueeze()

본 내용은 https://blog.naver.com/qbxlvnf11/221627488337 에서 대부분 발췌해 온 것임

 

 

Pytorch에서 텐서를 결합할 때는 cat() 함수를 활용,

dim 파라미터를 통해 기준을 설정

​

- torch 라이브러리 import

import torch

- 테스트용 텐서 생성

a = torch.ones(4, 3)
b = torch.zeros(3, 3)