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
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容