actually no, it currently must be one dimensional (its raveled earlier)
one could allow multidimensional which would mean partition each axis by a different kth, I'm not sure if thats worth the effort.