keras 获取某层的输入/输出 tensor 尺寸

最新推荐文章于 2023-11-03 21:01:26 发布

原创

最新推荐文章于 2023-11-03 21:01:26 发布 · 6k 阅读

2 ·

CC 4.0 BY-SA版权

本文介绍了如何在 Keras 中获取某层的输入和输出 Tensor 的尺寸。对于单输入层，当该层仅使用一次时，可以方便地获取其尺寸。若层被使用多次，只需稍作调整。对于输出 Tensor 的尺寸，通过替换为 'output' 即可查询。期待读者的反馈，无论是点赞、提问还是指出问题，都请友好交流。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

获取单输入尺寸，该层只被使用了一次。

import keras
from keras.layers import Input, LSTM, Dense, Conv2D
from keras.models import Model
a = Input(shape=(32, 32, 3))
b = Input(shape=(64, 64, 3))

conv = Conv2D(16, (3, 3), padding='same')
conved_a = conv(a)

# 到目前为止只有一个输入，以下可行：
assert conv.input_shape == (None, 32, 32, 3)

如果该层被使用了两次

import keras
from keras.layers import Input, LSTM, Dense, Conv2D
from keras.models import Model
a = Input(shape=(32, 32, 3))
b = Input(shape=(64, 64, 3))

conv = Conv2D(16, (3, 3), padding='same')
conved_a = conv(a)

# 到目前为止只有一个输入，以下可行：
assert conv.input_shape == (None, 32, 32, 3)

conved_b = conv(b)
# 现在 `.input_shape` 属性不可行，但是这样可以：
assert conv.get_input_shape_at(0) == (None, 32, 32, 3)
assert conv.get_input_shape_at(1) == (None, 64, 64, 3)

如果是输出，只需要改成output就好：