Ce este nn liniar în PyTorch?
Ce este nn liniar în PyTorch?

Video: Ce este nn liniar în PyTorch?

Video: Ce este nn liniar în PyTorch?
Video: Torch.nn.Linear Module explained 2024, Aprilie
Anonim

Din documentație: torță CLASS. nn . Liniar (in_features, out_features, bias=True) Se aplică a liniar transformarea la datele primite: y = xW^T + b. Parametri: in_features – dimensiunea fiecărei probe de intrare.

În mod similar, se întreabă, cum funcționează NN linear?

Liniar . Se aplică a liniar transformarea în datele primite, adică //y= Ax+b//. Tensorul de intrare dat în forward(input) trebuie să fie fie un vector (tensor 1D) fie o matrice (tensor 2D). Dacă intrarea este o matrice, atunci fiecare rând se presupune a fi un eșantion de intrare din lotul dat.

În mod similar, ce este conv2d în PyTorch? conv2d (input, greutate, sine. părtinire, sine. pas, sine. umplutură, sine. dilatare, sine.

În al doilea rând, ce este PyTorch nn?

PyTorch : nn The nn pachetul definește un set de module, pe care le puteți considera ca un strat de rețea neuronală care produce ieșire din intrare și poate avea unele greutăți antrenabile. lanterna de import # N este dimensiunea lotului; D_in este dimensiunea de intrare; # H este dimensiunea ascunsă; D_out este dimensiunea de ieșire.

Cum folosești ReLU în PyTorch?

În PyTorch , puteți construi a ReLU strat folosind funcția simplă relu1 = nn. ReLU cu argumentul inplace=False. De când ReLU funcția este aplicată în funcție de elemente, nu este nevoie să specificați dimensiunile de intrare sau de ieșire. Argumentul inplace determină modul în care funcția tratează intrarea.

Recomandat: