Method forward

examples/rnn/imdb_model.py:43–49

(self, x)

        self.l2 = layer.Linear(2)

    def forward(self, x):
        y = self.lstm(x)
        y = autograd.reshape(y, (y.shape[0], -1))
        y = self.l1(y)
        y = autograd.relu(y)
        y = self.l2(y)
        return y

    def train_one_batch(self, x, y):
        out = self.forward(x)