python显示数据长度_python-3.x – 在Tensorflow中获取数据集的长度

source_dataset = tf.data.TextLineDataset('primary.csv')

target_dataset = tf.data.TextLineDataset('secondary.csv')

dataset = tf.data.Dataset.zip((source_dataset, target_dataset))

dataset = dataset.shard(10000, 0)

dataset = dataset.map(lambda source, target: (tf.string_to_number(tf.string_split([source], delimiter=',').values, tf.int32),

tf.string_to_number(tf.string_split([target], delimiter=',').values, tf.int32)))

dataset = dataset.map(lambda source, target: (source, tf.concat(([start_token], target), axis=0), tf.concat((target, [end_token]), axis=0)))

dataset = dataset.map(lambda source, target_in, target_out: (source, tf.size(source), target_in, target_out, tf.size(target_in)))

dataset = dataset.shuffle(NUM_SAMPLES) #This is the important line of code

我想完全整理我的整个数据集,但是shuffle()需要拉取一些样本,而tf.Size()不能与tf.data.Dataset一起使用.

我怎么能正常洗牌？