rockpool.nn.combinators.residual

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

Functions

Residual(*args, **kwargs)

Build a residual block over a sequential stack of modules

Classes

JaxResidual(*args, **kwargs)	The Residual combinator for jax modules
ModResidual(*args, **kwargs)	The Residual combinator for native modules
ResidualMixin(*args, **kwargs)	The base class for the Residual combinator
TorchResidual(*args, **kwargs)	The Residual combinator for torch modules