tfl.premade_lib.build_linear_combination_layer

View source on GitHub

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

ensemble_outputs Ensemble outputs to be linearly combined.
model_config Model configuration object describing model architecture. Should be one of the model configs in tfl.configs.
dtype dtype

A tfl.layers.Linear instance.