代码片段记录18

class :
    def __next__(self):
        span_length = self.c * self.data_len
        rand_index = random.sample(list(range(math.ceil(span_length))), self.batch_size)
        sampled_sents = [self.data['sents'][i] for i in rand_index]
        sampled_labels = [self.data['labels'][i] for i in rand_index]
        batch = {'sents': sampled_sents, 'labels': sampled_labels}
        batch = convert2tensor(batch)
        return batch
    
    def __iter__(self):
        return self