(self, inputs)
| 90 | self.softmax_cross_entropy = layer.SoftMaxCrossEntropy() |
| 91 | |
| 92 | def forward(self, inputs): |
| 93 | y = self.linear1(inputs) |
| 94 | y = self.relu(y) |
| 95 | y = self.linear2(y) |
| 96 | return y |
| 97 | |
| 98 | def train_one_batch(self, x, y): |
| 99 | out = self.forward(x) |