(self)
| 37 | return output, hidden |
| 38 | |
| 39 | def init_hidden(self): |
| 40 | if torch.cuda.is_available(): |
| 41 | hidden = torch.zeros(self.n_layers, 1, self.hidden_size).cuda() |
| 42 | else: |
| 43 | hidden = torch.zeros(self.n_layers, 1, self.hidden_size) |
| 44 | |
| 45 | return Variable(hidden) |
| 46 | |
| 47 | |
| 48 | def str2tensor(string): |