Jaringan saraf pertama Anda
Sekarang saatnya Anda mengimplementasikan sebuah jaringan saraf kecil yang berisi dua layer linear yang disusun berurutan.
Latihan ini adalah bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Petunjuk latihan
- Tambahkan sebuah kontainer untuk menumpuk layer secara berurutan.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
import torch
import torch.nn as nn
input_tensor = torch.Tensor([[2, 3, 6, 7, 9, 3, 2, 1]])
# Create a container for stacking linear layers
model = nn.____(nn.Linear(8, 4),
nn.Linear(4, 1)
)
output = model(input_tensor)
print(output)