tf.data.Dataset.from_tensor_slices( )

我們在轉化數據集時經常會使用這個函數,他的所用是切分傳入的 Tensor 的第一個維度,生成相應的 dataset 。 1.對傳入的(5,2)進行切分,最終產生的dataset有5個元素,每個元素的形狀都是(2,) 2.在圖像識別中可能出現的字典或者元組的矩陣情況,因爲將圖像數字化之後,會產生矩陣和對應的標籤。比如: {」image」:image_tensor,」label」:label_tens
本站公眾號
   歡迎關注本站公眾號,獲取更多信息