Síť s lineární vrstvou
Neuronové sítě často obsahují mnoho vrstev, ale většina z nich jsou lineární vrstvy. Pochopení jedné lineární vrstvy ti pomůže zorientovat se v tom, jak fungují, než začneš přidávat další složitost.
Aplikuj lineární vrstvu na vstupní tensor a sleduj výstup.
Toto cvičení je součástí kurzu
Introduction to Deep Learning with PyTorch
Pokyny k cvičení
- Vytvoř vrstvu
Linear, která přijímá 3 příznaky na vstupu a vrací 2 výstupy. - Předej
input_tensorpřes lineární vrstvu.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
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)