tensorflow 添加一個全連接層


對於一個全連接層,tensorflow都為我們封裝好了。

使用:tf.layers.dense()

 1 tf.layers.dense(
 2     inputs,
 3     units,
 4     activation=None,
 5     use_bias=True,
 6     kernel_initializer=None,
 7     bias_initializer=tf.zeros_initializer(),
 8     kernel_regularizer=None,
 9     bias_regularizer=None,
10     activity_regularizer=None,
11     kernel_constraint=None,
12     bias_constraint=None,
13     trainable=True,
14     name=None,
15     reuse=None
16 )

inputs:  該層的輸入張量

units:  輸出節點的大小


免責聲明!

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



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