src/transformers/models/fnet/modeling_fnet.py/FNetBasicOutput/forward
class FNetBasicOutput:    def forward(self, hidden_states, input_tensor):
        hidden_states = self.LayerNorm(input_tensor + hidden_states)
        return hidden_states