keras 查看输入的shape
时间: 2024-04-30 08:21:16 浏览: 98
keras读取训练好的模型参数并把参数赋值给其它模型详解
在 Keras 中,可以使用 `model.summary()` 方法来查看模型的输入、输出形状以及各层的参数数量,例如:
```python
from keras.models import Sequential
from keras.layers import Dense
model = Sequential()
model.add(Dense(64, input_shape=(10,), activation='relu'))
model.add(Dense(1, activation='sigmoid'))
model.summary()
```
上述代码定义了一个包含两个全连接层的简单模型,第一层的输入形状为 `(batch_size, 10)`,其中 `batch_size` 表示每次训练时输入的样本数目。使用 `model.summary()` 可以打印出如下的模型结构信息:
```
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
dense_1 (Dense) (None, 64) 704
_________________________________________________________________
dense_2 (Dense) (None, 1) 65
=================================================================
Total params: 769
Trainable params: 769
Non-trainable params: 0
_________________________________________________________________
```
其中 `Output Shape` 列就是每层的输出形状,第一层的输出形状为 `(None, 64)`,其中 `None` 表示不确定的 `batch_size`,第二层的输出形状为 `(None, 1)`。
阅读全文