tf.data.Dataset.map()

最新推荐文章于 2025-10-30 14:11:00 发布

原创最新推荐文章于 2025-10-30 14:11:00 发布 · 2.4k 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

当前文章被以下社区和专栏收录：

这篇博客介绍了如何利用TensorFlow的tf.data.Dataset.map函数对数据集进行映射操作，通过定义函数fun将数据集中的每个元素加上1。示例中展示了如何从一个包含数据和索引的字典数据集出发，应用自定义函数获取新的数据集，并遍历输出结果。

Python3.8

Python 是一种高级、解释型、通用的编程语言，以其简洁易读的语法而闻名，适用于广泛的应用，包括Web开发、数据分析、人工智能和自动化脚本

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

在这里插入图片描述