rockpool.nn.combinators.ffwd_stack

Implement a combinator that creates feed-forward module stacks, by placing a linear module in between each module

Functions

FFwdStack(*args, **kwargs)

Assemble modules into a feed-forward stack, with linear weights in between

Classes

FFwdStackMixin(*args, **kwargs)	Assemble modules into a feed-forward linear stack, with linear weights in between
JaxFFwdStack(*args, **kwargs)
ModFFwdStack(*args, **kwargs)
TorchFFwdStack(*args, **kwargs)