code: def model_action_q(self):
          
          q = self.get('model:q')
          actions = self.get('rollout:actions')
          return q.gather(1, actions.unsqueeze(1)).squeeze(1)
docs: Action values for selected actions in the rollout