fix bug with multi-dimensional inputs for KGENP KGENP_D

because `xt::sort` does not behave nicely with series containing NaN
values (e.g. it does not consistently put NaN values at the end/
beginning), so need to eliminate them before the sorting, which requires
to treat each series separately