ntakouris · September 25, 2020 14:56
diff --git a/flat_map_no_perf.py b/flat_map_no_perf.py
 # no parallel read performance improvement compared to interleave
 dataset = tf.data.Dataset.range(2).flat_map(lambda x: tf.data.Dataset.range(x*3, (x + 1 )* 3))

 for t in dataset.take(5):
  print(t.numpy())

 # 0
 # 1
 # 2
 # 3
 # 4
	# no parallel read performance improvement compared to interleave
	dataset = tf.data.Dataset.range(2).flat_map(lambda x: tf.data.Dataset.range(x3, (x + 1 ) 3))

	for t in dataset.take(5):
	print(t.numpy())

	# 0
	# 1
	# 2
	# 3
	# 4