1. 程式人生 > >tensorflow 新增一個全連線層

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:  輸出節點的大小