Python numpy 및 tensor의 차원 열벡터, 행벡터 인덱싱

개요

Python의 numpy와 tensor를 indexing할 때 차원의 축과 행 방향으로 indexing 하는 방법을 알아보자.

 

Numpy와 tensor의 indexing

먼저 아래와 같은 (2,2,5) shape을 가진 numpy의 배열이 있다고 가정하자.

필자는 이 배열을 표현할 때 다음과 같이 표현하려고 한다.

 

"2개의 배치와 2개의 행 그리고 5개의 열을 가진 배열"

 

그렇다면 이때, 각 배치의 첫 번째 행벡터들을 indexing 하려면 어떻게 해야 할까?

 

1. 행벡터 indexing

그 방법은 바로 아래와 같은 코드를 작성하면 된다.

그러면 아래와 같은, 우리가 원했던 indexing을 수행할 수 있다.

 

2. 열벡터 indexing

그렇다면, 각 배치의 축 방향 벡터들, 열벡터들을 indexing 하려면 어떻게 해야 할까?

그 방법은 바로 아래와 같은 코드를 작성하면 된다.

그러면 아래와 같은, 우리가 원했던 indexing을 수행할 수 있다.