KerasRS / API 文档 / 嵌入层 / FeatureConfig 配置类

FeatureConfig 配置类

[源代码]

FeatureConfig

keras_rs.layers.FeatureConfig(
    name: str,
    table: keras_rs.src.layers.embedding.distributed_embedding_config.TableConfig,
    input_shape: Sequence[Optional[int]],
    output_shape: Sequence[Optional[int]],
)

一个嵌入特征的配置。

keras_rs.layers.DistributedEmbedding 配置一个特征。每个特征使用通过 keras_rs.layers.TableConfig 配置的表,并且多个特征可以共享同一个表。

属性

  • name: 特征的名称。必须定义。
  • table: 用于查找此特征的表。
  • input_shape: 特征的输入形状。馈送到层的特征必须与该形状匹配。请注意,对于输入中的不规则维度,此处提供的维度表示最大值;任何更大的值都将被截断。
  • output_shape: 特征激活的输出形状。嵌入层返回的内容必须与该形状匹配。