def maxprob(self, prob):
        '''
            Method deterministically sample actions of maximum likelihood
        '''
        if len(prob.shape) == 3:
            return prob[:, :, self.d]
        return prob[:, :self.d]

            Method deterministically sample actions of maximum likelihood
        
surreal/model/ppo_net.py