MLPLayer

MLPLayer#

class nfflr.nn.MLPLayer(in_features: int, out_features: int, norm: Literal['layernorm', 'batchnorm'] = 'layernorm')[source]#

Multilayer perceptron layer helper.

Methods

forward(x)

Linear, norm, silu layer.

forward(x)[source]#

Linear, norm, silu layer.