应该更正为: r_out = r_out.view(-1, 64) outs = self.out(r_out) return outs, h_state