Pytorch 在张量中找到唯一向量答案

【问题标题】：Pytorch find unique vectors in tensorPytorch 在张量中找到唯一向量
【发布时间】：2022-01-20 15:54:18
【问题描述】：

我有一个包含二进制值的张量，例如

T1 = torch.tensor([[1., 0., 1.],
    [0., 1., 0.],
    [1., 0., 1.]])

我需要将其转换为：

tensor([[1., 0., 1.],
        [0., 1., 0.]])

我查看了torch.unique，但它只适用于价值观？

有没有办法对整个向量进行唯一操作

【问题讨论】：

我不确定我是否理解您要查找的结果，这些行的排序依据是什么？您能否为输入 T2 = torch.tensor([[1., 0., 1.], [0., 1., 1.], [1., 0., 1.]]) 提供所需的结果？
结果：torch.tensor([[1., 0., 1.], [0., 1., 1.]])
为什么第二行会有三个元素，因为第 n°3 列只有 1，这是寻找唯一值的副作用吗？
我正在寻找唯一的行，而不是唯一的列值。所以第 1 行与第 3 行相同，因此我只返回其中的 1 个

标签： pytorch

【解决方案1】：

虽然 PyTorch 文档不是很清楚 dim 参数可以实现这一点，但我发现这篇文章解决了问题 Delete duplicated rows in torch.tensor

因此

torch.unique(T1,dim=0)

解决问题

【讨论】：