flambe.nn.mlp
¶
Module Contents¶
-
class
flambe.nn.mlp.
MLPEncoder
(input_size: int, output_size: int, n_layers: int = 1, output_activation: Optional[nn.Module] = None, hidden_size: Optional[int] = None, hidden_activation: Optional[nn.Module] = None)[source]¶ Bases:
flambe.nn.module.Module
Implements a multi layer feed forward network.
This module can be used to create output layers, or more complex multi-layer feed forward networks.
-
seq
¶ the sequence of layers and activations
Type: nn.Sequential
-