tf.data.Dataset介绍1-from_tensor_slices

本文主要是介绍tf.data.Dataset介绍1-from_tensor_slices，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

1、tf.data.Dataset.from_tensor_slices使用

a.对输入的类型异常敏感，遇到第一个为list或tensor类型作为维度作为切割维度。小数据内存处理时使用。
b.对于外层tuple或dict的跳过不作为切割维度（可以视为多个tensor的列表或类似表现形式）。

  c.tensor中每一个特征的向量维度要保持一致，且维度必须是大于等于1

例如，list数据格式slice

#list
list(tf.data.Dataset.from_tensor_slices([[1,3,4],[2,3,4]]).as_numpy_iterator())Out[10]: [array([1, 3, 4]), array([2, 3, 4])]#tuple
list(tf.data.Dataset.from_tensor_slices(([1,3,4],[2,3,4])).as_numpy_iterator())
Out[11]: [(1, 2), (3, 3), (4, 4)]#dict
list(tf.data.Dataset.from_tensor_slices({"a":[1,2,3],"b":[5,6,7]}).as_numpy_iterator())
Out[12]: [{'a': 1, 'b': 5}, {'a': 2, 'b': 6}, {'a': 3, 'b': 7}]#tensor
list(tf.data.Dataset.from_tensor_slices(tf.constant([[1,2,3],[4,5,6]])).as_numpy_iterator())
Out[13]: [array([1, 2, 3]), array([4, 5, 6])]

参考：tf.data.Dataset  |  TensorFlow Core v2.8.0 

这篇关于tf.data.Dataset介绍1-from_tensor_slices的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---