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)