获得keras神经网络某一层输出的维度等信息

如下述的网络结构时:

def test_model(input):
    input=keras.layers.Input(input)
    x=keras.layers.Conv2D(32,(5,5),strides=(1,1),padding='same')(input)
    return keras.models.Model(input,x)
model=test_model((7,7,3))

只输出维度信息:

x=model.layers[1].output_shape
print(x)

获得第一层的输出:

x=model.get_layer(index=1).output
print(x)

输出如下:

或者

x=model.layers[1].output
print(x)

输出如下:

猜你喜欢

转载自blog.csdn.net/weixin_42074867/article/details/88547288
今日推荐