1. 程式人生 > >tf.contrib.layers.xavier_initializer

tf.contrib.layers.xavier_initializer

xavier_initializer(
    uniform=True,
    seed=None,
    dtype=tf.float32
)
  • 1
  • 2
  • 3
  • 4
  • 5

該函式返回一個用於初始化權重的初始化程式 “Xavier” 。

這個初始化器是用來保持每一層的梯度大小都差不多相同。

引數:

uniform: 使用uniform或者normal分佈來隨機初始化。 
seed: 可以認為是用來生成隨機數的seed 
dtype: 只支援浮點數。

返回值:

初始化權重矩陣