torch_unique_consecutive function

Unique_consecutive

Unique_consecutive

torch_unique_consecutive( self, return_inverse = FALSE, return_counts = FALSE, dim = NULL )

Arguments

  • self: (Tensor) the input tensor
  • return_inverse: (bool) Whether to also return the indices for where elements in the original input ended up in the returned unique list.
  • return_counts: (bool) Whether to also return the counts for each unique element.
  • dim: (int) the dimension to apply unique. If NULL, the unique of the flattened input is returned. default: NULL

TEST

Eliminates all but the first element from every consecutive group of equivalent elements.

.. note:: This function is different from [`torch_unique`] in the sense that this function
    only eliminates consecutive duplicate values. This semantics is similar to `std::unique`
    in C++.

Examples

if (torch_is_installed()) { x = torch_tensor(c(1, 1, 2, 2, 3, 1, 1, 2)) output = torch_unique_consecutive(x) output torch_unique_consecutive(x, return_inverse=TRUE) torch_unique_consecutive(x, return_counts=TRUE) }
  • Maintainer: Daniel Falbel
  • License: MIT + file LICENSE
  • Last published: 2025-02-14