Method argsort

mllm/core/Tensor.cpp:397–400

Tensor Tensor::abs() { return Context::instance().buildOpAndSubmitTask(OpTypes::kAbs, aops::AbsOpOptions{}, {*this})[0]; }

Tensor Tensor::argsort(int dim, bool descending) {
  return Context::instance().buildOpAndSubmitTask(OpTypes::kArgsort,
                                                  aops::ArgsortOpOptions{.dim = dim, .descending = descending}, {*this})[0];
}

Tensor Tensor::clip(float min_val, float max_val) {
  return Context::instance().buildOpAndSubmitTask(OpTypes::kClip, aops::ClipOpOptions{.min_val = min_val, .max_val = max_val},