`build_linear_stack`

python

build_linear_stack(*, create_linear: Callable[[int, int], nn.Module], input_features: int, hidden_sizes: list[int], activation: type[nn.Module]) -> tuple[list[nn.Module], int]

Build a stack of linear layers with activations.

Parameters:

create_linear (Callable[[int, int], Module]) – Factory used to create each linear layer.
input_features (int) – Number of input features to the first layer.
hidden_sizes (list[int]) – Hidden layer sizes.
activation (type[Module]) – Activation module type to apply after each hidden layer.

Returns:

tuple[list[Module], int] – (layers, output_features).