[pytorch] 파라미터 dim = ?의 의미...

tensor뿐만 아니라 pandas에서 concat을 할때 어떤식으로 진행할지를 결정하게 될때 사용하는게 dim 이라는 파라미터이다.

 

예제로 봐보자.

x,y,z를 stack하는 코드이다.

stack을 진행할때, 어떤 차원을 기준으로 stack을 할지 명시해줘야한다.

defalut는 0으로 dim = 0이라는 것과 같은 의미이다.

dim = 0일때 결과를 살펴보면, 행으로 stack한 것을 볼 수 있다.

반대로, dim= 1 일때를 살펴보면, '열'로 stack을 진행한 것을 볼 수 있다.

 

추가적으로 설명해보자면, torch.Size([A,B,C,D,E])와 같은 5차원의 tensor가 존재할 경우,

A부터 0, B는 1 , C는 2, D는 3, E는 4를 뜻한다.

이걸 알고 원하는 기준으로 stack을 진행하면 된다.