tf.keras.Input


tf.keras.Input用於實例化Keras tensor:

 

例如有a,b,c分別是Keras tensor,那么可以這樣寫到模型里:model = Model(input=[a, b], output=c) 。參數:

shape:tuple類型,不包含批量維度,例如shape=(32,)表明輸入為32-d的向量所組成的批量,可以為None,表示未知維度。

name:層的名字,可選。若要用的話應當唯一,如果不提供則自動生成。

dtype:輸入類型(float32float64int32...)

sparse:表明將要創建的placeholder是否是sparse。’ragged‘和’sparse‘僅有一個為True

tensor:可選,將已有的tensor來放入到Input層。如果設置的話則該層將不創建placeholder tensor

ragged:一個布爾值,指定要創建的placeholder是否是ragged。“ragged”和“sparse”中只有一個是真的。在這種情況下,“shape”參數中的“None”值表示不規則的維度。有關ragged tensors的更多信息,請參見https://www.tensorflow.org/guide/ragged_tensors

 

例子:

 

附: TF2資源:

簡單粗暴 TensorFlow 2.0 | A Concise Handbook of TensorFlow 2.0       wiki

TensorFlow 2.0案例實戰

30天吃掉那只 TensorFlow2.0 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM