Sobes.tech
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)