[PyTorch] torch view, squeeze, unsqueeze

view

주어진 모양대로 배열을 reshape한다.

 

Squeeze

dim이 1인 경우를 없애준다.

 

 

 

unsqueeze의 인자로 -1을 넣어보겠습니다. -1은 인덱스 상으로 마지막 차원을 의미합니다. 현재 크기는 (3,)이었으므로 마지막 차원에 1인 차원을 추가하면 (3, 1)의 크기를 가지게 됩니다. 다시 말해 현재 텐서의 경우에는 1을 넣은 경우와 -1을 넣은 경우가 결과가 동일합니다. 

 

 

 

 

 

 

위키독스

강의↓

파이토치로 시작하는 딥러닝 기초