AI 정리

Pytorch

dawon-project 2025. 3. 24. 23:20

Pytorch framework

  • 신경망 구축에 사용되는 소프트웨어 기반 오픈 소스 딥러닝 프레임워크
  • Torch의 머신 러닝 라이브러리 + Python 기반의 고급 API를 결합한 것

Pytorch vs Tensorflow(TF)

  Pytorch Tensorflow
개발사 Meta(Pytorch Foundation) Google
Ecosystem 연구, 학계 산업, 제품 배포
강점 사전 학습 모델과 문서가 많음 Google 서비스와의 통합에 편리

 

Pytorch 사용법 -> dtype 메서드를 통해 변수가 torch 자료형인지 확인하는 것이 중요

  • 텐서(Tensor) -> 파이토치 데이터 자료구조
    • PyTorch의 핵심 데이터 구조
    • Numpy 배열(ndarray)과 유사하지만 GPU 가속이 가능
    • 딥러닝의 계산을 위한 자동 미분이 가능하다 
    • (H, W, C)의 데이터 구조를 가지므로 아래의 예시에서 (2,2,3) 은 H : 2 / W : 2 / C : 3 이다.

  • PyTorch 자료형
    • torch.FloatTensor -> 32비트 실수형 텐서
    • torch.IntTensor -> 32비트 정수형 텐서
    • torch.cuda.FloatTensor -> GPU 가속된 실수형 텐서