lovely-tensors 라이브러리(https://xl0.github.io/lovely-tensors/) 써보셨나요? (저만 몰랐던 것일 수도... 😅)
Raschka 박사가 2022년 베스트 오픈소스 중 하나로 꼽아서 알게 된 것인데 사용하기 좋네요.
파이토치 텐서 x를 출력하면 아래와 같이 숫자값 뭉치가 나오잖아요?
tensor([[[-0.3541, -0.3369, -0.4054, ..., -0.5596, -0.4739, 2.2489], ...
'lovely-tensors'를 설치 및 활성화하고 출력하면 아래와 같이 값의 범위, 통계량 등 좀 더 사람이 알아보기 쉽게 나옵니다.
tensor[3, 196, 196] n=115248 x∈[-2.118, 2.640] μ=-0.388 σ=1.073
거기에 텐서 x에 대해 x.rgb라고 체이닝 메서드를 실행하면 바로 이미지를 볼 수 있고 x.plt를 실행하면 히스토그램을 바로 그려줍니다. 힘들게 matplotlib 불러오고 그럴 필요 없이요.
자매품으로 lovely-jax와 lovely-numpy도 있습니다. ❤️