AI/pytorch
Tensor 변환과 결합 - cat(), view(), squeeze(), unsqueeze()
DOAD
2021. 6. 28. 20:57
본 내용은 https://blog.naver.com/qbxlvnf11/221627488337 에서 대부분 발췌해 온 것임
Pytorch에서 텐서를 결합할 때는 cat() 함수를 활용,
dim 파라미터를 통해 기준을 설정
- torch 라이브러리 import
import torch
- 테스트용 텐서 생성
a = torch.ones(4, 3)
b = torch.zeros(3, 3)