model.py 文件源码

def _decode_one_step(self, y, pc, p, fb_mat, fbe_mat):
    j = self.y_j(_mkivar(y))
    q = self._context(p, fb_mat, fbe_mat)
    pc, p = F.lstm(pc, self.j_p(j) + self.q_p(q) + self.p_p(p))
    z = self.p_z(p)
    return z, pc, p