def sample_batch_transitions(self, batch_size, forward_steps=<NUM_LIT:1>):
batches = [<EOL>self._sample_batch_prioritized(<EOL>segment_tree, batch_size, self.deque.frame_history, forward_steps=forward_steps<EOL>)<EOL>for segment_tree in self.segment_trees<EOL>]<EOL>probs, idxs, tree_idxs = zip(*batches)<EOL>return np.stack(probs, axis=<NUM_LIT:1>).astype(float), np.stack(idxs, axis=<NUM_LIT:1>), np.stack(tree_idxs, axis=<NUM_LIT:1>)<EOL>
Return indexes of next sample