我的代码非常慢,因此我试图使其更快。我多次使用tf.data.Dataset.map()方法进行数据扩充:
dataset_1 = tf.data.Dataset.from_tensor_slices(example)
if bool_1 :
dataset_2 = dataset_1.map(fn_1_t)
else :
dataset_2 = dataset_1.map(fn_1_f)
if bool_2 :
dataset_3 = dataset_1.map(fn_2_t)
else :
dataset_3 = dataset_1.map(fn_2_f)
final_dataset = dataset_3.map(fn_3)
您是否认为使用多个映射不是一个好主意,是否会使CPU无法进行并行计算?
谢谢!