所有的Keras層對(duì)象都有如下方法:
layer.get_weights()
:返回層的權(quán)重(numpy array)
layer.set_weights(weights)
:從numpy array中將權(quán)重加載到該層中,要求numpy array的形狀與* layer.get_weights()
的形狀相同
layer.get_config()
:返回當(dāng)前層配置信息的字典,層也可以借由配置信息重構(gòu)
from keras.utils.layer_utils import layer_from_configconfig = layer.get_config()layer = layer_from_config(config)
如果層僅有一個(gè)計(jì)算節(jié)點(diǎn)(即該層不是共享層),則可以通過(guò)下列方法獲得輸入張量、輸出張量、輸入數(shù)據(jù)的形狀和輸出數(shù)據(jù)的形狀:
layer.input
layer.output
layer.input_shape
layer.output_shape
如果該層有多個(gè)計(jì)算節(jié)點(diǎn)(參考層計(jì)算節(jié)點(diǎn)和共享層)??梢允褂孟旅娴姆椒?/p>
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)
聯(lián)系客服