함수1 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 | Pyt.. 2022. 6. 23. 이전 1 다음 반응형