Method __init__

beginner_source/introyt/modelsyt_tutorial.py:50–56

(self)

class TinyModel(torch.nn.Module):
    
    def __init__(self):
        super().__init__()
        
        self.linear1 = torch.nn.Linear(100, 200)
        self.activation = torch.nn.ReLU()
        self.linear2 = torch.nn.Linear(200, 10)
        self.softmax = torch.nn.Softmax(dim=1)
    
    def forward(self, x):
        x = self.linear1(x)