我可以提取pytorch张量中对应于某个键的所有索引吗？

Question

假设我有一个pytorch张量。我想提取3出现的索引，即tensor([0,3,5])。这有内置的功能吗

Answer 1

t = torch.Tensor([1, 2, 3 , 2 , 5])
print ((t == 2).nonzero())

非零打印火炬张量的所有非零位置 https://pytorch.org/docs/master/generated/torch.nonzero.html

Answer 2

有一个专用的function用于：

   torch.where(my_tensor == the_number)