โครงข่ายประสาทเทียมแรกของคุณ
ถึงเวลาสร้างโครงข่ายประสาทเทียมขนาดเล็กที่ประกอบด้วยเลเยอร์แบบ linear สองเลเยอร์เรียงต่อกัน
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
Deep Learning เบื้องต้นด้วย PyTorch
คำแนะนำการฝึกหัด
- เพิ่มคอนเทนเนอร์สำหรับเรียงซ้อนเลเยอร์ต่อกันเป็นลำดับ
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
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)