Add a norm keyword arg to fft functions

This fixes #2142.
I prefer `a.shape[axis]` but that is just me.