rockpool.nn.modules.native.instant

Encapsulate a simple instantaneous function as a jax module

Classes

Instant(*args,Β **kwargs)

Wrap a callable function as an instantaneous Rockpool module

InstantJax(*args,Β **kwargs)

Wrap a callable function as an instantaneous Rockpool module, with a Jax backend

InstantMixin(shape,Β function,Β *args,Β **kwargs)

Wrap a callable function as an instantaneous Rockpool module

InstantTorch(*args,Β **kwargs)

Wrap a callable function as an instantaneous Rockpool module, with a Torch backend