torch_unique_consecutive: Unique_consecutive

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Unique_consecutive

Description

Unique_consecutive

Usage

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)
}

torch documentation built on May 29, 2024, 9:54 a.m.