去创作
新版本：
class SampleEpoch(): def __init__(self, dataset): self.dataset = dataset self.size = np.array(self.dataset).shape[0] self.ind = np.array([], dtype='int64') def __call__(self, batch_size): if self.ind.shape[0] < batch_size: self.ind = np.hstack((self.ind, np.random.permutation(np.arange(self.size)))) ... ...
