PyTorchPyTorch

import torch
import torch.nn as nn

# 텐서 생성
x = torch.tensor([1.0, 2.0, 3.0])
y = x ** 2  # [1, 4, 9]

# GPU 사용
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
x = x.to(device)

# 자동 미분
x = torch.tensor(3.0, requires_grad=True)
y = x ** 2
y.backward()   # dy/dx 계산
print(x.grad)  # tensor(6.) — 2*3=6

# 모델 정의
class Net(nn.Module):
    def __init__(self):
        super().__init__()
        self.fc = nn.Linear(10, 1)

    def forward(self, x):
        return self.fc(x)