1. 程式人生 > >tf.data.Dataset.from_tensor_slices( )

tf.data.Dataset.from_tensor_slices( )

我們在轉化資料集時經常會使用這個函式,他的所用是切分傳入的 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] }的形式。