tf.keras学习之input

参考

tensorflow社区

做什么的?

对输入做一个初始化,自动生成的。

使用

tf.keras.Input(
shape=None,
batch_size=None,
name=None,
dtype=None,
sparse=False,
tensor=None,
ragged=False,
**kwargs
)

参数

shape:shape=(32,)输入将是32维向量的。None代表形状未知的尺寸。
batch_size:批量大小
name:起一个名字。唯一的。默认会自动生成。
dtype:数据类型(float32,float64,int32…)
sparse:占位符是否稀疏。“ ‘ragged’ 和’sparse’ 中只有一个可以是True。
tensor:可自定义现有tensor,如不定义将自动创建占位符。
ragged:占位符是否参差不齐。在这种情况下,“shape”自变量中的“None”值表示参差不齐的尺寸。
有关RaggedTensors的一些内容。
**kwargs:不推荐使用,支持batch_shape和 batch_input_shape。

例子

x = Input(shape=(16,)) # 输入
y = Dense(8, activation='softmax')(x) # 输出
model = Model(x, y)
#通过输入和输出生成中间的model

猜你喜欢

转载自blog.csdn.net/fuzizhu1/article/details/106102630