This allows complex sorting based on a key, which can be the abs, real, imag
etc of the value array. So the user can choose which metric they want to use.

* Sort now allows all dimensions
* Sort if much faster by using batched mode. This takes up more memory though.
* Enabled large tests for sort
* Added tests for sorting on dim1 and dim2