生成一维张量作为2D张量行的独特索引

Question

假设我们通过给出每个行不同索引，从0 行 - 行数-1，我们将2D张量转换为1D张量。

[[1,2],[1,3],[1,4]] -> [0,1,2]

但是，如果有相同的行，那么我们将重复索引，如下所示。

[[1,2],[1,2],[1,4]] -> [0,0,2]

[[1,2],[1,3],[1,2]] -> [0,1,0]

如何在Pytorch上实施此操作？

Answer 1

您可以使用 /a>并返回为框外提供索引的反向：

>>> _, i = x.unique(dim=0, return_inverse=True)

>>> i # first example
tensor([0, 1, 2])