test/linalg/norm: added a reference function supporting custom axes and generalised the test
linalg/norm: using the correct output shape in case of non-default axes

Fixes #13.