Input() 用于实例化 Keras 张量。
输入参数:
- shape: 一个尺寸元组(整数),不包含批量大小。 例如,shape=(32,) 表明期望的输入是按批次的 32 维向量。
- batch_shape: 一个尺寸元组(整数),包含批量大小。 例如,batch_shape=(10, 32) 表明期望的输入是 10 个 32 维向量。
- batch_shape=(None, 32) 表明任意批次大小的 32 维向量。
- name: 一个可选的层的名称的字符串。 在一个模型中应该是唯一的(不可以重用一个名字两次)。 如未提供,将自动生成。
- dtype: 输入所期望的数据类型,字符串表示 (float32, float64, int32…)
- sparse: 一个布尔值,指明需要创建的占位符是否是稀疏的。
tensor: 可选的可封装到 Input 层的现有张量。 如果设定了,那么这个层将不会创建占位符张量。
返回参数:
- 一个张量
示例:
from keras.layers import Input, Dense
from keras.models import Model
x = Input(shape=(32, ))
y = Dense(16, activation = 'softmax')(x)
model = Model(x, y)
model.summary()
# Model: "model_1"
# _________________________________________________________________
# Layer (type) Output Shape Param #
# =================================================================
# input_1 (InputLayer) (None, 32) 0
# _________________________________________________________________
# dense_1 (Dense) (None, 16) 528
# =================================================================
# Total params: 528
# Trainable params: 528
# Non-trainable params: 0
# _________________________________________________________________