当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.nn.RNNCellDeviceWrapper.set_weights用法及代码示例


用法

set_weights(
    weights
)

参数

  • weights NumPy 数组的列表。数组的数量及其形状必须与层权重的维数相匹配(即,它应该与 get_weights 的输出相匹配)。

抛出

  • ValueError 如果提供的权重列表与图层的规格不匹配。

从 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)]

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.nn.RNNCellDeviceWrapper.set_weights。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。