def METHOD_NAME(self, src, seg, tgt):
    emb = self.embedding(src, seg)
    memory_bank = self.encoder(emb, seg)
    emb = self.tgt_embedding(tgt, None)
    hidden = self.decoder(memory_bank, emb, (src,))
    output = self.target.lm.output_layer(hidden)
    return output