(self, layer)
| 391 | super(RNN, self).__init__(*args, **kwargs) |
| 392 | |
| 393 | def create_cell(self, layer): |
| 394 | if layer == 0: |
| 395 | input_size = self.input_size |
| 396 | else: |
| 397 | input_size = self.num_directions * self.hidden_size |
| 398 | return RNNCell(input_size, self.hidden_size, self.bias, self.nonlinearity) |
| 399 | |
| 400 | def init_hidden(self, batch_size): |
| 401 | hidden_shape = ( |