PyTorch Guide 🔥🤖

Deep Learning & AI Framework

1. What is PyTorch?

PyTorch is an open-source deep learning framework used to build neural networks and AI models.

2. Installation

pip install torch torchvision

3. Import

import torch
import torch.nn as nn

4. Tensors

x = torch.tensor([1, 2, 3])
print(x)

5. Neural Network

model = nn.Sequential(
  nn.Linear(1, 10),
  nn.ReLU(),
  nn.Linear(10, 1)
)

6. Loss Function

loss_fn = nn.MSELoss()

7. Optimizer

optimizer = torch.optim.Adam(model.parameters(), lr=0.01)

8. Training Loop

for epoch in range(10):
  y_pred = model(X)
  loss = loss_fn(y_pred, y)

  optimizer.zero_grad()
  loss.backward()
  optimizer.step()

9. Real Use Cases