(self, hidden, encoder_output)
| 754 | self.v = nn.Parameter(torch.FloatTensor(hidden_size)) |
| 755 | |
| 756 | def dot_score(self, hidden, encoder_output): |
| 757 | return torch.sum(hidden * encoder_output, dim=2) |
| 758 | |
| 759 | def general_score(self, hidden, encoder_output): |
| 760 | energy = self.attn(encoder_output) |