Add a norm keyword arg to fft functions

This fixes #2142.
A bit annoying maybe, but it would be nice to have tests for all of the functions (maybe you can just loop over them all). To make sure nobody accidentally removes the support.