Action values for selected actions in the rollout

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