(self, config)
| 835 | """ Compute SQuAD start_logits from sequence hidden states. """ |
| 836 | |
| 837 | def __init__(self, config): |
| 838 | super().__init__() |
| 839 | self.dense = nn.Linear(config.hidden_size, 1) |
| 840 | |
| 841 | def forward(self, hidden_states, p_mask=None): |
| 842 | """ Args: |