언어
PyTorch 빠른 참조
PyTorch 빠른 참조용 인쇄 가능한 치트시트로, 빠른 조회와 학습, 책상 옆 참고용으로 정리되었습니다
미리보기
이 치트시트에 포함된 내용
- 텐서
- 자동 미분
- 신경망
- 데이터 로딩
- 학습 루프
- 옵티마이저
- 손실 함수
- 저장 및 로딩
- GPU
- 일반 패턴
텐서 생성
import torch
a = torch.tensor([1, 2, 3])
b = torch.zeros(2, 3)
c = torch.ones(3, 3)
d = torch.randn(2, 4) # normal dist
텐서 생성자
torch.zeros(m, n) | 모두 0, 형태 (m, n) |
torch.ones(m, n) | 모두 1, 형태 (m, n) |
torch.randn(m, n) | 표준 정규 분포 난수 |
torch.arange(start, end, step) | 균등 간격 값 |
torch.linspace(start, end, steps) | 고정 개수의 점 |
torch.eye(n) | 단위 행렬 |
torch.empty(m, n) | 초기화되지 않은 메모리 |
NumPy 상호 운용
t = torch.from_numpy(np_array)
arr = tensor.numpy() # shares memory
t = torch.as_tensor(np_array)
모든 섹션이 포함된 인쇄용 치트시트 다운로드
PDF 다운로드다운로드
