sampling.py 文件源码

def get_next_sample(self):
        if self.incomplete_samples_ is not None:
            assert self.incomplete_samples_.size > 0
            split = np.vsplit(self.incomplete_samples_, [1])
            assert len(split) == 2
            next_sample = split[0]
            if split[1].size == 0:
                self.incomplete_samples_ = None
            else:
                self.incomplete_samples_ = split[1]
            return next_sample.ravel()
        else:
            return None