FeedForward

FeedForward#

class nfflr.nn.FeedForward(d_in: int, d_hidden: int | None = None, d_out: int | None = None, norm: bool = False)[source]#

Two-layer feedforward network.

Methods

forward(x)

Doc for forward.

reset_parameters

forward(x: Tensor)[source]#

Doc for forward.