Python tf.nn.RNNCellDeviceWrapper.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)]