Method slice

training/verl/protocol.py:122–125

(self, index)

        return DataProtoItem(batch=tensor_data, non_tensor_batch=non_tensor_data, meta_info=self.meta_info)

    def slice(self, index):
        tensor_data = self.batch[index]
        non_tensor_data = {key: val[index] for key, val in self.non_tensor_batch.items()}
        return DataProto(batch=tensor_data, non_tensor_batch=non_tensor_data, meta_info=self.meta_info)

    def slice_batch(self, start, length, dim=0):
        """