Quelle est la définition de classe de nn.Linéaire dans Pytorch

Question

J'ai le code suivant pour 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

Ma question: qu'est-ce que c'est self.hidden?

Il revient de nn.Linear Et peut prendre x comme argument. Quelle est exactement la fonction de self.hidden?

Merci

Answer 1

Le Network défini comme ayant deux couches, caché et sortie. En gros, la fonction de la couche cachée est de contenir des paramètres que vous pouvez optimiser pendant l'entraînement.