pytorch 공부 (1) - 기본 연산

연산

`torch`의 연산은 `numpy`와 동일하게 작동한다.

import torch
import numpy as np

a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
a + b # tensor([5, 7, 9])


a * 3 # tensor([3, 6, 9])

 

 

a = torch.tensor(3)
a.item() # 3
type(a.item()) # int

- `.item()` 함수를 통해 scalar tensor의 값을 뽑아낸다. (값이 여럿일 경우에는 오류가 난다.)

 

자동 미분

# AutoGrad
x = torch.tensor(1.0, requires_grad=True) # AutoGrad 적용

a = x*3
y = a **2 # (3x)**2 = 9x**2 -> 미분하면 18x
y.backward() # 어떤 값에 대한 출력 tensor의 변화도를 전달 받고, 동일한 값에 대한 입력 tensor의 변화도를 계산

x.grad # tensor(18.)