Return indexes of next random rollout

def sample_batch_trajectories(self, rollout_length):
        """ Return indexes of next random rollout """
        results = []

        for i in range(self.num_envs):
            results.append(self.sample_rollout_single_env(rollout_length))

        return np.stack(results, axis=-1)