fastdev.nn.mlp ============== .. py:module:: fastdev.nn.mlp Module Contents --------------- .. py:class:: MLP(input_dim: int, output_dim: int, hidden_dims: List[int], activation_layer: Optional[Type[torch.nn.Module]] = nn.ReLU, activation_on_output: bool = False, residual_on_output: bool = False, residual_on_hidden: bool = False, use_normalization: bool = False, normalization_layer: Optional[Type[torch.nn.Module]] = nn.LayerNorm) Bases: :py:obj:`torch.nn.Module` A flexible MLP .. py:attribute:: dims .. py:attribute:: residual_on_hidden :value: False .. py:attribute:: residual_on_output :value: False .. py:attribute:: layers .. py:method:: forward(x: torch.Tensor) -> torch.Tensor .. py:data:: mlp