tf.data.Dataset.from_tensor_slices( )

2022-10-27 08:36:29

我们在转化数据集时经常会使用这个函数,他的所用是切分传入的 Tensor 的第一个维度,生成相应的 dataset 。

1.对传入的(5,2)进行切分,最终产生的dataset有5个元素,每个元素的形状都是(2,)

2.在图像识别中可能出现的字典或者元组的矩阵情况,因为将图像数字化之后,会产生矩阵和对应的标签。比如:

{”image”:image_tensor,”label”:label_tensor}的形式。

函数会分别切分”a”中的数值以及”b”中的数值,最后总dataset中的一个元素就是类似于{ “a”:1.0, “b”:[0.9,0.1] }的形式。

  • 作者:Kun Li
  • 原文链接:https://blog.csdn.net/u012193416/article/details/83720078
    更新时间:2022-10-27 08:36:29