flambe.nn.sequential

Module Contents

class flambe.nn.sequential.Sequential(**kwargs: Dict[str, Union[Module, torch.nn.Module]])[source]

Bases: flambe.nn.Module

Implement a Sequential module.

This class can be used in the same way as torch’s nn.Sequential, with the difference that it accepts kwargs arguments.

forward(self, data: torch.Tensor)[source]

Performs a forward pass through the network.

Parameters:data (torch.Tensor) – input to the model
Returns:output – output of the model
Return type:torch.Tensor