Xếp chồng các lớp tuyến tính

Bạn đã làm tốt khi xây dựng mạng đầu tiên với hai lớp tuyến tính. Giờ hãy xếp thêm vài lớp nữa. Hãy nhớ rằng một mạng nơ-ron có thể có bao nhiêu lớp ẩn tùy ý, miễn là đầu vào và đầu ra khớp với nhau.

Mạng này được thiết kế để nhận đầu vào sau:

input_tensor = torch.Tensor(
    [[2, 7, 9, 5, 3]]
    )