src/transformers/models/fnet/modeling_fnet.py/FNetBasicOutput/__init__
class FNetBasicOutput:    def __init__(self, config):
        super().__init__()
        self.LayerNorm = nn.LayerNorm(config.hidden_size, eps=config.layer_norm_eps)