Python wrappers
Public Python APIs implemented by this engine symbol.Return the indices that would sort a along axis.
Parameters
aTensorImplPtrInput tensor.
axisintAxis along which to compute the sort permutation.
Returns
TensorImplPtrInteger tensor of the same shape as a and dtype I32, such that a.take_along_axis(out, axis) equals sort_op(a, axis).
Notes
Non-differentiable (indices are discontinuous in the input values).