PyTorch Framework

CuriHuS·2024년 6월 12일
0

AI & Game Programming

목록 보기
6/8

Deep Learning Frameworks

  • 딥러닝 프레임워크의 초점
    • 새로운 아이디어의 빠른 프로토타입 가능
    • 자동적으로 gradient 계산
    • GPU에서 효율적으로 실행됨

PyTorch

근본적인 컨셉

  • Tensor: numpy 배열처럼 GPU위에서 실행됨 = 자료형
  • Autograd: Tensor의 계산 그래프를 패키징하고 자동으로 gradient를 계산해줌
  • Module: 뉴런 네트워크 레이어; 상태나 가중치를 저장해줌

Tensor

Autograd

  • Tensor 생성 시에 requires_grad=True로 바꿔 autograd를 활성화 시켜줌
  • Tensor에서 계산은 PyTorch가 계산 그래프를 만들게 유도함






New Functions

nn

profile
공부 노트

0개의 댓글

관련 채용 정보