Python tf.keras.layers.Layer.get_weights用法及代碼示例

用法

get_weights()

返回層的當前權重，作為 NumPy 數組。

層的權重表示層的狀態。此函數將與該層關聯的可訓練和不可訓練權重值作為 NumPy 數組列表返回，該數組又可用於將狀態加載到類似參數化的層中。

例如，Dense 層返回兩個值的列表：內核矩陣和偏置向量。這些可用於設置另一個 Dense 層的權重：

layer_a = tf.keras.layers.Dense(1,
  kernel_initializer=tf.constant_initializer(1.))
a_out = layer_a(tf.convert_to_tensor([[1., 2., 3.]]))
layer_a.get_weights()
[array([[1.],
       [1.],
       [1.]], dtype=float32), array([0.], dtype=float32)]
layer_b = tf.keras.layers.Dense(1,
  kernel_initializer=tf.constant_initializer(2.))
b_out = layer_b(tf.convert_to_tensor([[10., 20., 30.]]))
layer_b.get_weights()
[array([[2.],
       [2.],
       [2.]], dtype=float32), array([0.], dtype=float32)]
layer_b.set_weights(layer_a.get_weights())
layer_b.get_weights()
[array([[1.],
       [1.],
       [1.]], dtype=float32), array([0.], dtype=float32)]