Deep Learning study

[Pytorch] torch.Tensor.view 본문

AI/Pytorch api

[Pytorch] torch.Tensor.view

HwaniL.choi 2018. 5. 23. 20:30
반응형
view(*args)  Tensor


view는 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