src/transformers/models/fnet/modeling_fnet.py/FNetFourierTransform/forward
class FNetFourierTransform:    def forward(self, hidden_states):
        self_outputs = self.self(hidden_states)
        fourier_output = self.output(self_outputs[0], hidden_states)
        outputs = (fourier_output,)
        return outputs