[python] numpy array, torch tensor 크기 확인하기

가끔 헷갈릴 때가 있어서 크기 확인 방법을 한 번에 정리해보자!

 

1. numpy array 함수들: shape, size

그림1

• shape: 행렬의 각 차원 크기를 반환한다.
• size: 행렬의 전체 사이즈, 즉 원소 개수를 반환한다.

np.shape는 아래와 같이 torch.tensor에 대해서도 사용할 수 있다.

• np.shape(tensor)

그림2. np.shape(tensor) 결과

역시 각 차원의 길이를 반환한다.

 

 

2. Torch tensor

그림3. a는 그림2와 같은 torch tensor 타입

• tensor.shape == np.shape(tensor): 각 차원의 길이, 즉 shape 반환
• a.size(0) == a.shape[0]: 차원 0의 길이 반환
• a.size(1) == a.shape[1]: 차원 1의 길이 반환