def sample_batch_transitions(self, batch_size, forward_steps=<NUM_LIT:1>):
results = []<EOL>for i in range(self.num_envs):<EOL><INDENT>results.append(self.sample_frame_single_env(batch_size, forward_steps=forward_steps))<EOL><DEDENT>return np.stack(results, axis=-<NUM_LIT:1>)<EOL>
Return indexes of next sample