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:  输出节点的大小