Reinitialize learnable parameters.
(self)
| 57 | |
| 58 | |
| 59 | def reset_parameters(self): |
| 60 | """ |
| 61 | Reinitialize learnable parameters. |
| 62 | """ |
| 63 | nn.init.xavier_normal_(self.W) |
| 64 | nn.init.xavier_normal_(self.a) |
| 65 | |
| 66 | |
| 67 | def _get_attention_scores(self, h_transformed: torch.Tensor): |