Norm

Contents

Norm#

class nfflr.nn.Norm(num_features: int, norm_type: Literal['batchnorm', 'layernorm', 'instancenorm'] | None = 'layernorm', mode: Literal['node', 'edge'] = 'node')[source]#

Methods

forward(x)