.cast( )函數的使用

tf.cast( ) 或者K.cast( ) 是執行 tensorflow 中的張量數據類型轉換,比如讀入的圖片是int8類型的,一定要在訓練的時候把圖片的數據格式轉換爲float32.這樣就能夠將其轉換成 0 或 1 的序列,進行one-hot encoding。
相關文章
相關標籤/搜索