MulaiMulai sekarang secara gratis

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

Lihat Kursus

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)
Edit dan Jalankan Kode