[3주차 부캠]pytorch 필수과제 1

index_select()

index_select(tensor, 차원, 뽑을거) >> 차원 : 기준이 되는 차원 (0이면 y 기준 1이면 x기준) / 뽑을거 : 뽑을 index들

dim 파라미터를 0,1 중에 어떤걸 선택하느냐에 따라서 아래 그림에서 파랑, 빨강이 같이 뽑을 방향이 결정된다.