pytorchのnn.Linearのクラス定義は何ですか

Question

私はpytorchの次のコードを持っています：

import torch.nn as nn
import torch.nn.functional as F

class Network(nn.Module):
    def __init__(self):
        super().__init__()
        self.hidden = nn.Linear(784, 256)
        self.output = nn.Linear(256, 10)

    def forward(self, x):
        x = F.sigmoid(self.hidden(x))
        x = F.softmax(self.output(x), dim=1)

        return x

私の質問：これは何ですかself.hidden？

nn.Linearから戻り、xを引数として取ることができます。 self.hiddenの機能は正確には何ですか？

ありがとう

Answer 1

Networkは、非表示と出力の2つのレイヤーがあると定義されています。大まかに言えば、非表示層の機能は、トレーニング中に最適化できるパラメーターを保持することです。