def METHOD_NAME(self, src, seg):
    emb = self.embedding(src, seg)
    output = self.encoder(emb, seg)
    output = self.act(self.target.linear_1(output))
    output = self.target.layer_norm(output)
    output = self.target.linear_2(output)
    prob = torch.nn.Softmax(dim=-1)(output)
    return prob