tensorly.sort

sort(tensor, axis, descending=False)

Return a sorted copy of an array

Parameters
tensortensor

An N-D tensor

axisint or None

Axis along which to sort. If None, the array is flattened before sorting. The default is -1, which sorts along the last axis.

descendingbool

If True, values are sorted in descending order, otherwise in ascending.

Returns
sorted_tensortensor

An N-D array, sorted copy of input tensor