rockpool.nn.combinators.sequential

Implement the Sequential combinator, with helper classes for Jax and Torch backends

Functions

Sequential(*args, **kwargs)

Build a sequential stack of modules by connecting them end-to-end

Classes

JaxSequential(*args, **kwargs)	The Sequential combinator for Jax modules
ModSequential(*args, **kwargs)	The Sequential combinator for native modules
SequentialMixin(*args, **kwargs)	Base class for Sequential modules
TorchSequential(*args, **kwargs)	The Sequential combinator for torch modules