获得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)

 

输出如下:

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值