tf.data.Dataset.map()

tf.data.Dataset.map(fun)：map的参数是一个函数fun

import tensorflow as tf
def fun(x):
    return x +1
ds = tf.data.Dataset.range(5)
ds = ds.map(fun)

tf.data.Dataset.map()函数_nofish_xp的博客-CSDN博客_dataset map函数 

map()：接受函数fun，Dataset中的每个数据都会被作为这个函数的参数进行计算输出，组成新的数据集 Dataset

dataset = tf.data.Dataset.from_tensor_slices({'data':np.array([1.0, 2.0, 3.0, 4.0, 5.0]),'index':np.random.uniform(size=(5,3))})
 
def get_new_dataset(dataset):
    data = dataset['data']
    index = dataset['index']
    return  data, index
 
dataset = dataset.map(get_new_dataset)
for data, index in dataset:
    print(data,"===",index)
    print("---")