[DL] Pytorch Tutorial

Pytorch Tutorial. 텐서의 정의 및 크기변경 등

설치 및 import

!pip install torch
import torch

텐서의 초기화

torch.empty(5, 3)  # 임의 초기화,  5 by 3 
torch.zeros(5, 3, dtype = torch.long) # 0으로 초기화, 5 by 3
torch.ones(5, 3, dtype = torch.long) # 1로 초기화, 5 by 3

텐서의 초기화(random 활용)

torch.randn(5, 3) # 정규분포 기준
torch.rand(5, 3) # 0~1 사이의 균등분포

다차원 텐서도 가능함. (ex - 3차원)

torch.randn(4, 3, 3)
# 이 경우 3 * 3 행렬이 4층으로 쌓이는 구조가 된다.

텐서의 크기 변경

x = torch.randn(4, 4)
y = x.view(16) # 1차원 벡터 16개 성분짜리로 바뀐다
z = y.view([-1, 2]) # 하나의 크기를 모를경우 -1로 하면 자동으로 바뀐다. 8 by 2가 된다.