Sieć z warstwą liniową
Sieci neuronowe często składają się z wielu warstw, z których większość to warstwy liniowe. Zrozumienie działania pojedynczej warstwy liniowej pozwoli ci pojąć zasadę ich działania, zanim przejdziesz do bardziej złożonych struktur.
Zastosuj warstwę liniową do tensora wejściowego i sprawdź wynik.
To ćwiczenie jest częścią kursu
Wprowadzenie do uczenia głębokiego z PyTorch
Instrukcje do ćwiczenia
- Utwórz warstwę
Linear, która przyjmuje 3 cechy na wejściu i zwraca 2 wartości na wyjściu. - Przepuść
input_tensorprzez warstwę liniową.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
import torch
import torch.nn as nn
input_tensor = torch.tensor([[0.3471, 0.4547, -0.2356]])
# Create a Linear layer
linear_layer = nn.____(
in_features=____,
out_features=____
)
# Pass input_tensor through the linear layer
output = ____(____)
print(output)