def _decode_one_step(self, y, qc, q): j = self.y_j(_mkivar(y)) qc, q = F.lstm(qc, self.j_q(j) + self.q_q(q)) z = self.q_z(q) return z, qc, q