1. 程式人生 > >tf.layers.dense()層的定義

tf.layers.dense()層的定義

dense(
    inputs,
    units,
    activation=None,
    use_bias=True,
    kernel_initializer=None,
    bias_initializer=tf.zeros_initializer(),
    kernel_regularizer=None,
    bias_regularizer=None,
    activity_regularizer=None,
    trainable=True,
    name=None,
    reuse=None
)

 

引數:

inputs:tf.layers.dense 的Tensor輸入。
units:整數或長整數,輸出空間的維數。
activation:啟用功能(可呼叫),將其設定為“None”以保持線性啟用。
use_bias:Boolean,表示該層是否使用偏差。
kernel_initializer:權重矩陣的初始化函式;如果為None(預設),則使用tf.get_variable使用的預設初始化程式初始化權重。
bias_initializer:偏置的初始化函式。
kernel_regularizer:權重矩陣的正則化函式。
bias_regularizer:正規函式的偏差。
activity_regularizer:輸出的正則化函式。
kernel_constraint:由Optimizer更新後應用於核心的可選投影函式(例如,用於實現層權重的範數約束或值約束)。該函式必須將未投影的變數作為輸入,並且必須返回投影變數(必須具有相同的形狀)。在進行非同步分散式訓練時,使用約束是不安全的。
bias_constraint:由Optimizer更新後應用於偏置的可選投影函式。
trainable:Boolean,如果為True,還將變數新增到圖集合GraphKeys.TRAINABLE_VARIABLES中(請參閱參考資料tf.Variable)。
name:String,圖層的名稱;具有相同名稱的圖層將共享權重,但為了避免錯誤,在這種情況下,我們需要reuse=True。
reuse:Boolean,是否以同一名稱重用前一層的權重。