Frobenius norm is defined also in the `elif len(axis) == 2:` branch below, so there doesn't seem to be a need to define it on top.