tf.data.Dataset.from_tensor_slices的用法

语义解释:from_tensor_slices,从张量的切片读取数据。

工作原理:将输入的张量的第一个维度看做样本的个数,沿其第一个维度将tensor切片,得到的每个切片是一个样本数据。实现了输入张量的自动切片。

输入数据格式/要求:
1)可以是numpy格式,也可以是tensorflow的tensor的格式,函数会自动将numpy格式转为tensorflow的tensor格式
2)输入可以是一个tensor
或 一个tensor字典(字典的每个key对应的value是一个tensor,要求各tensor的第一个维度相等)
或 一个tensor tuple(tuple 的每个元素是一个tensor,要求各tensor的第一个维度相等)。


示例1
示例2:传入tensor
示例3:传入元组
示例4:传入字典
示例5:传入元组-2

你可能感兴趣的:(tf.data.Dataset.from_tensor_slices的用法)