Linear(num_hidden, 1) self.num_hidden = num_hidden def forward(self, img): flattened = img.view(-1, 28 * 28) activation1 = self.layer1(flattened) ...
確定! 回上一頁