搜索 data/values 的张量

Question

鉴于：

tensor([[6, 6],
        [4, 8],
        [7, 5],
        [7, 4],
        [6, 4]])

如何找到值为 [7,5] 的行的索引？

一般来说，如何搜索任何值的索引：完整和部分行或列？

Answer 1

试试这个：

>>> (a[:, None] == torch.tensor([7, 5])).all(-1).any(-1).nonzero().flatten().item()
2
>>>