argmax axis

argmax axis를 자꾸 헷갈리는데,

[   ,   ,   ],

[   ,   ,   ]

전체를 놓고 봐야 함. 그리고 벡터는 열벡터로 시작한다는 것을 잊지 말아야 함.(열 다음 행 , 0 다음 1)

 

axis = 0이면 (전체를 놓고 봤을 때) 한 열에서 가장 큰 인덱스 반환

[   ,   ,   ],

[   ,   ,   ]

   표시 중에서 큰 것을 선택

 

axis = 1이면 (전체를 놓고 봤을 때) 한 행에서 가장 큰 인덱스 반환

[   ,   ,   ],

[   ,   ,   ]

   표시 중에서 큰 것을 선택

 

그러니까 가장 안쪽에 있는 배열부터 선택된다고 생각하면 된다고 생각했는데

3차원이면 굉장히 복잡함 ..