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("---")
