밑바닥부터 시작하는 딥러닝 5

156

밑바닥부터 시작하는 딥러닝 5

설치가 완료되면 다음 코드를 실행합니다. 파이토치 버전이 출력되면 제대로 설치된 것입니다.

import torch

#

버전

출력

print(torch

.

__

version

__

) # 2

.

x

.

x

이 책에서는 파이토치

2

.

0

.

0

이상을 사용합니다. 단, 기본적인 기능만 사용하므로 다른 버전에

서도 그대로 동작하거나 약간만 수정하면 문제를 바로잡을 수 있을 것입니다.

6.1.2

텐서 계산

설치를 끝냈으니 사용해봅시다.

tensor

클래스부터 시작하겠습니다.

tensor

는 텐서, 즉 다차

원 배열을 표현한 클래스입니다. 넘파이의

np

.

ndarray

와 똑같이 취급할 수 있습니다. 다음 코

드를 예시로 실행해봅시다.

import torch

x

=

torch

.

tensor(5

.

0)

y

=

3

*

x

**

2

print(y)

실행 결과

tensor(75

.

)

보다시피 넘파이와 마찬가지로 조작법이 직관적입니다. 이제 미분을 구해보겠습니다. 앞의 코

드에서는

y

=

3

*

x

**

2

라는 계산을 했습니다. 이 계산의

x

=

5

에서 미분을 구해봅시다. 먼저

수식으로 답을 구해보겠습니다. 이번 계산은 수식으로

이며 미분은 입니다.

를 대입하면 정답은

30

입니다.

코드에서도

30

이 나오는지 확인해봅시다. 파이토치에서 미분은

tensor

인스턴스의

back

ward

( )