Python tf.nn.RNNCellResidualWrapper.set_weights用法及代碼示例

用法

set_weights(
    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)]