Function unique_

tfjs-core/src/ops/unique.ts:78–89

* Finds unique elements along an axis of a tensor. * * It returns a tensor `values` containing all of the unique elements along the * `axis` of the given tensor `x` in the same order that they occur along the * `axis` in `x`; `x` does not need to be sorted. It also returns a tensor * `indices`

(
    x: T|TensorLike, axis = 0)

 * @doc {heading: 'Operations', subheading: 'Evaluation'}
 */
function unique_<T extends Tensor>(
    x: T|TensorLike, axis = 0): {values: T, indices: Tensor1D} {
  const $x = convertToTensor(x, 'x', 'unique', 'string_or_numeric');
  assert($x.rank > 0, () => 'The input tensor must be at least 1D');

  const inputs: UniqueInputs = {x: $x};
  const attrs: UniqueAttrs = {axis};
  const [values, indices] = ENGINE.runKernel(
      Unique, inputs as unknown as NamedTensorMap,
      attrs as unknown as NamedAttrMap) as [T, Tensor1D];
  return {values, indices};
}

export const unique = /* @__PURE__ */ op({unique_});