`tolist()` 함수는 파이썬의 리스트로 데이터를 변환하는 메서드입니다. 이 함수는 다차원 배열 또는 텐서를 일차원 리스트로 변환하여 반환합니다.
일반적으로, NumPy 배열이나 PyTorch의 텐서와 같은 다차원 데이터 구조에서 `tolist()` 함수를 사용하면 원소들이 한 차원 아래로 펼쳐진 리스트로 변환됩니다. 예를 들어, 2차원 NumPy 배열을 `tolist()` 함수로 변환하면, 각 행이 리스트로 변환되고 이 리스트들이 하나의 리스트로 묶여 반환됩니다.
예시를 통해 설명하겠습니다. 다음은 2차원 NumPy 배열을 `tolist()` 함수를 사용하여 리스트로 변환하는 예입니다:
import numpy as np arr = np.array([[1, 2, 3], [4, 5, 6]]) lst = arr.tolist() print(lst) # 출력: [[1, 2, 3], [4, 5, 6]]
`tolist()` 함수를 사용하면 `arr` 배열이 리스트 `lst`로 변환되었습니다. 각 행이 하나의 리스트로 변환되고, 이 리스트들이 하나의 리스트로 묶여 반환됩니다.
마찬가지로, PyTorch의 텐서에서도 `tolist()` 함수를 사용하여 텐서를 리스트로 변환할 수 있습니다. 결과는 동일한 원리로 다차원 텐서가 리스트로 변환됩니다.
`tolist()` 함수는 다차원 배열 또는 텐서를 파이썬 리스트로 변환할 때 유용하게 사용됩니다.