[PyTorch] 사칙연산

해당 글은 pytorch에서 tensor의 사칙연산 4가지를 다룬다.

 

0. torch 라이브러리 선언 및 tensor 객체 생성

import torch
A = torch.Tensor([5])
B = torch.Tensor([10])

1. 덧셈

print(A+B)
print(torch.add(A, B))

tensor([15.])

tensor([15.])

 

 

2. 뺄셈

print(A-B)
print(torch.sub(A, B))

tensor([-5.])

tensor([-5.])

 

 

3. 곱셈

print(A*B)
print(torch.mul(A, B))

tensor([50.])

tensor([50.])

 

 

4. 나눗셈

print(A/B)
print(torch.div(A, B))

tensor([0.5000])

tensor([0.5000])