dataset.py 文件源码

def rejection_resample(self, ds):
    nclasses = 1000
    def _classfunc(*tensors):
      as_dict = self.dictify(tensors)
      uids = as_dict['uid']
      return tf.mod(uids, nclasses)
    target_dist = tf.constant(1/nclasses, shape=(nclasses,))
    return tf.contrib.data.rejection_resample(ds, _classfunc, target_dist)