PyTorch torch.unique_consecutive function
Pytorch torch Reference Manual
torch.unique_consecutiveIt is a function in PyTorch used to obtain consecutive unique values. It returns adjacent consecutive unique values in the tensor.
Function Definition
torch.unique_consecutive(input, return_inverse=False, return_counts=False, dim=None)
Usage Examples
Example
import torch
# Create tensor
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# Get consecutive unique values
unique_vals = torch.unique_consecutive(x)
print(f"Consecutive unique values: {unique_vals}")
# Return counts
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Counts: {counts}")
# Create tensor
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# Get consecutive unique values
unique_vals = torch.unique_consecutive(x)
print(f"Consecutive unique values: {unique_vals}")
# Return counts
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Counts: {counts}")
Other Extensions