Keras 3 API 文档 / 层 API / 核心层 / 输入对象

输入对象

[源]

Input 函数

keras.Input(
    shape=None,
    batch_size=None,
    dtype=None,
    sparse=None,
    ragged=None,
    batch_shape=None,
    name=None,
    tensor=None,
    optional=False,
)

用于实例化 Keras 张量。

Keras 张量是一个符号张量对象,我们为其添加了某些属性,以便仅通过了解模型的输入和输出就可以构建 Keras 模型。

例如,如果 abc 是 Keras 张量,则可以这样做:model = Model(input=[a, b], output=c)

参数

  • shape: 形状元组(整数或 None 对象的元组),不包括批次大小。例如,shape=(32,) 表示预期输入将是 32 维向量的批次。此元组的元素可以是 NoneNone 元素表示形状未知且可能变化的维度(例如序列长度)。
  • batch_size: 可选的静态批次大小(整数)。
  • dtype: 输入预期的数据类型,字符串形式(例如 "float32""int32"...)
  • sparse: 布尔值,指定预期输入是否为稀疏张量。请注意,如果 sparseFalse,仍然可以将稀疏张量传递到输入中 - 它们将被密集化,默认值为 0。此功能仅支持 TensorFlow 和 JAX 后端。默认为 False
  • ragged: 布尔值,指定预期输入是否为不规则(ragged)张量。请注意,如果 raggedFalse,仍然可以将不规则张量传递到输入中 - 它们将被密集化,默认值为 0。此功能仅支持 TensorFlow 后端。默认为 False
  • batch_shape: 可选的形状元组(整数或 None 对象的元组),包括批次大小。
  • name: 可选的层名称字符串。在模型中应唯一(不要两次使用相同的名称)。如果未提供,将自动生成。
  • tensor: 可选的现有张量,用于封装到 Input 层中。如果设置,层将使用此张量而不是创建新的占位符张量。
  • optional: 布尔值,指示输入是否可选。可选输入可以接受 None 值。

返回

一个 Keras 张量。

示例

# This is a logistic regression in Keras
x = Input(shape=(32,))
y = Dense(16, activation='softmax')(x)
model = Model(x, y)