Python PyTorch unique_consecutive用法及代码示例

本文简要介绍python语言中 torch.unique_consecutive 的用法。

用法:

torch.unique_consecutive(*args, **kwargs)

参数：

• input(Tensor) -输入张量

• return_inverse(bool) -是否还返回原始输入中的元素在返回的唯一列表中结束的位置的索引。

• return_counts(bool) -是否还返回每个唯一元素的计数。

• dim(int) -应用唯一的维度。如果 None ，则返回平展输入的唯一性。默认值：None

返回：

一个张量或一个张量元组包含

• 输出(Tensor)：唯一标量元素的输出列表。

• inverse_indices(Tensor)：(可选)如果return_inverse为 True，将有一个额外的返回张量(与输入的形状相同)表示原始输入中的元素映射到输出中的位置的索引；否则，此函数将仅返回一个张量。

• 计数(Tensor)：(可选)如果return_counts为 True 时，将有一个额外的返回张量(与输出或 output.size(dim) 相同的形状，如果指定了dim)，表示每个唯一值或张量的出现次数。

返回类型：

(Tensor、Tensor(可选)、Tensor(可选))

从每个连续的等效元素组中消除除第一个元素之外的所有元素。

注意

此函数与 torch.unique() 的不同之处在于此函数仅消除连续重复值。此语义类似于 C++ 中的std::unique。

例子：

>>> x = torch.tensor([1, 1, 2, 2, 3, 1, 1, 2])
>>> output = torch.unique_consecutive(x)
>>> output
tensor([1, 2, 3, 1, 2])

>>> output, inverse_indices = torch.unique_consecutive(x, return_inverse=True)
>>> output
tensor([1, 2, 3, 1, 2])
>>> inverse_indices
tensor([0, 0, 1, 1, 2, 3, 3, 4])

>>> output, counts = torch.unique_consecutive(x, return_counts=True)
>>> output
tensor([1, 2, 3, 1, 2])
>>> counts
tensor([2, 2, 1, 2, 1])