ML 라이브러리
Why Pytorch
Pytorch
다차원 Arrays를 표현하는 Pytorch 클래스
사실상 numpy의 ndarray와 동일 (TensorFlow의 Tensor와도 동일)
Tensor를 생성하는 함수도 거의 동일
import torch
import numpy as np
n_array = np.arrange(10).reshape(2,5)
test_tensor = torch.FloatTensor(n_array)
data = [[3, 5], [10, 5]]
x_data = torch.tensor(data)
## tip ##
x.data = x.data.to('cuda') # tensor는 GPU에 올려서 사용가능
Tensor handling and operation
view: reshape과 동일하게 tensor의 shape을 변환
squeeze: 차원의 개수가 1인 차원을 삭제 (압축)
unsqueeze: 차원의 개수가 1인 차원을 추가
행렬곱셈 연산은 dot이 아닌 mm, matmul 사용
nn.functional 모듈: 다양한 수식 변환을 지원 (one_hot....)
AutoGrad
Pytorch 프로젝트 Template
Module 구성