Deep Learning study
[Pytorch] torch.Tensor.view 본문
반응형
view
(*args) → Tensorview는 Tensor의 size를 바꾸어주는 역할을 한다 .
이건 다른것 보다는 예시를 보는게 이해가 빠를것 같다.
y라는 tensor를 하나 준다.
y를 y.size(1) * y.size(2) * y.size(0) 로 view한다.
결과는 아래와 같다.
마찬가지로 2*2*3 이였던 tensor를 3*4 tensor로 바꾸었다.
위의 y tensor를 4*3 tensor로 바꾸었다.
view를 하는 방식은 tensor안에있는 값들을 차례대로 나열하여 다시 재배열 시키는 것 같다.
위의 y로 예를들면
1 1 2 2 3 3 4 4 5 5 6 6 을 4*3으로 만들려고 하면
1 1 2 / 2 3 3 / 4 4 5 / 5 6 6 이런식으로 자르는 것이다.
그리고 아직도 헷갈리는게 axis개념인데 내가 수학공부할때의 것과 좀 상이한 부분이 있는것 같다고 해야되나.. 여튼 헷갈린다. 그것도 다시 공부해야겠다.
반응형
'AI > Pytorch api' 카테고리의 다른 글
[Pytorch]torch.nn.functoinal.pad() (2) | 2020.12.29 |
---|---|
[Pytorch] torch.nn.Sequential (0) | 2018.04.02 |
[Pytorch] torch.nn.Upsample (0) | 2018.04.01 |
[Pytorch] torch.nn.Conv2d (0) | 2018.04.01 |
Comments