Middle
Ați lucrat cu PyTorch?
sobes.tech AI
Răspuns de la AI
Da, am lucrat cu PyTorch. Este o bibliotecă populară pentru învățarea profundă, care permite crearea și antrenarea rețelelor neuronale cu un grafic de calcul dinamic. Am folosit-o pentru sarcini de clasificare a imaginilor și procesare a limbajului natural, aplicând modulele torch.nn pentru construirea modelelor și torch.optim pentru optimizare. Exemplu de un model foarte simplu:
import torch
import torch.nn as nn
class SimpleNet(nn.Module):
def __init__(self):
super().__init__()
self.fc = nn.Linear(10, 1)
def forward(self, x):
return self.fc(x)
model = SimpleNet()
input_tensor = torch.randn(5, 10)
output = model(input_tensor)
print(output)