python size function

github에서 LAS를 구현한 코드를 보다가 아래 코드를 발견했다.

seq_lengths    = [s[0].size(1) for s in batch]

size()함수는 tensor의 크기나 array의 크기를 알 수 있는 함수다. 

그런데 size(1)이라고 1이라는 인자가 입력으로 들어가는 것을 볼 수 있다.

나는 이런걸 처음 봐서 한참을 헤매었는데, 공식문서는 아니지만 아래 페이지에서 답을 찾을 수 있었다.

바로 axis, 즉 축 번호를 나타낸다.

0을 넣으면 행 크기, 1을 넣으면 열 크기, 더 큰 수는 고차원의 축에 대해 크기를 반환해 준다.

---

참고: https://www.geeksforgeeks.org/numpy-size-function-python/

Numpy size() function | Python - GeeksforGeeks