목록Machine Learning/Pytorch (25)
elevne's Study Note

PyTorch 는 2017 년에 공개된 딥러닝 라이브러리로 Lua 언어로 개발되었던 Torch 를 페이스북에서 파이썬 버전으로 내놓은 것이다. PyTorch 는 GPU 에서 텐서 조작 및 동적 신경망 구축을 지원한다. 내부적으로 CUDA, cuDNN 를 사용하여 GPU 연산을 진행한다. 아래 사진은 PyTorch 의 구조를 보여준다. Top-level API 에 해당하는 패키지들은 아래와 같다. torch: GPU 를 지원하는 텐서 패키지로 다차원 텐서를 기반으로 다양한 수학적 연산이 가능하도록 한다. torch.autograd: 자동 미분 패키지로, '자동 미분(auto-differentiation)' 이라는 기술을 사용하여 미분 계산을 효율적으로 처리한다. '연산 그래프'가 즉시 계산(실시간으로 네트..
Machine Learning/Pytorch
2023. 1. 5. 22:16