copynet_generation_score.py

class CopyNetSeq2Seq(Model):
    
    # snip...
    
    def _get_generation_scores(self, state: Dict[str, torch.Tensor]) -> torch.Tensor:
        # `self._output_generation_layer` is just a PyTorch linear layer with an input
        # dimension equal to the decoder hidden state size, and an output dimension
        # equal to the size of the target vocabulary.
        return self._output_generation_layer(state["decoder_hidden"])