view
주어진 모양대로 배열을 reshape한다.
Squeeze
dim이 1인 경우를 없애준다.
unsqueeze의 인자로 -1을 넣어보겠습니다. -1은 인덱스 상으로 마지막 차원을 의미합니다. 현재 크기는 (3,)이었으므로 마지막 차원에 1인 차원을 추가하면 (3, 1)의 크기를 가지게 됩니다. 다시 말해 현재 텐서의 경우에는 1을 넣은 경우와 -1을 넣은 경우가 결과가 동일합니다.
위키독스
온라인 책을 제작 공유하는 플랫폼 서비스
wikidocs.net
강의↓
파이토치로 시작하는 딥러닝 기초
부스트코스 무료 강의
www.boostcourse.org
'AI > PyTorch' 카테고리의 다른 글
| [PyTorch] Linear Regression 작성하기 (0) | 2021.06.01 |
|---|---|
| [PyTorch] Tensor와 NumPy (0) | 2021.06.01 |
| [PyTorch] package (0) | 2021.05.24 |
| [pyTorch] 이미지 여는 방법 (0) | 2021.05.12 |
| [PyTorch] torch 함수 (0) | 2021.04.27 |