code: def sample_batch_trajectories(self, rollout_length):
          
          results = []
  
          for i in range(self.num_envs):
              results.append(self.sample_rollout_single_env(rollout_length))
  
          return np.stack(results, axis=-1)
docs: Return indexes of next random rollout