网络编程
位置:首页>> 网络编程>> Python编程>> Keras设置以及获取权重的实现

Keras设置以及获取权重的实现

作者:silent56_th  发布时间:2021-11-22 10:04:09 

标签:Keras,设置,权重

layer的两个函数:

get_weights(), set_weights(weights)。

详情请参考about-keras-layers。

补充知识:Keras层的共同函数

关于Keras层:

所有Keras层都有很多共同的函数:

layer.get_weights(): # 以Numpy矩阵的形式返回层的权重。

layer.set_weights(weights): # 从Numpy矩阵中设置层的权重(与get_weights的输出形状相同)。

layer.get_config(): # 返回包含层配置的字典。

图层的重置:


layer = Dense(32)
config = layer.get_config()
reconstructed_layer = Dense.from_config(config)
#
from keras import layers

config = layer.get_config()
layer = layers.deserialize({'class_name': layer.__class__.__name__,
      'config': config})

如果一个层具有单个节点, (i.e. 如果它不是共享层), 可以得到它的输入张量,输出张量,输入尺寸和输出尺寸:


layer.input
layer.output
layer.input_shape
layer.output_shape

如果层有多个节点 (层节点和共享层), 可以使用以下函数: 要指明再哪个节点处获得张量,哪个节点处获得张量尺寸。


layer.get_input_at(node_index)
layer.get_output_at(node_index)
layer.get_input_shape_at(node_index)
layer.get_output_shape_at(node_index)

来源:https://blog.csdn.net/silent56_th/article/details/61663548

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com