線形レイヤーを積み重ねる

2つの線形レイヤーからなる最初のネットワークを作成できましたね。では、さらにレイヤーを積み重ねてみましょう。入力と出力の整合が取れていれば、ニューラルネットワークの隠れ層は必要なだけ増やせます。

このネットワークは次の入力を受け取るように設計されています。

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