tfl.premade_lib.build_linear_layer

Creates a tfl.layers.Linear layer initialized to be an average.

Main aliases

tfl.premade.premade_lib.build_linear_layer

tfl.premade_lib.build_linear_layer(
    linear_input,
    feature_configs,
    model_config,
    weighted_average,
    submodel_index,
    dtype
)

Args
`linear_input`	Input to the linear layer.
`feature_configs`	A list of `tfl.configs.FeatureConfig` instances that specify configurations for each feature.
`model_config`	Model configuration object describing model architecture. Should be one of the model configs in `tfl.configs`.
`weighted_average`	If the linear coefficients should be positive and sum up to one.
`submodel_index`	Corresponding index into submodels.
`dtype`	dtype

Returns
A `tfl.layers.Linear` instance.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-08-02 UTC.

Args