Jaringan saraf pertama Anda
Sekarang saatnya Anda mengimplementasikan sebuah jaringan saraf kecil yang berisi dua layer linear yang disusun berurutan.
Latihan ini merupakan bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Instruksi latihan
- Tambahkan sebuah kontainer untuk menumpuk layer secara berurutan.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
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)