fix bug where self.last_forward_log_prob was only storing last table instead of sum