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

假设我有一个pytorch张量

张量（[3,5,7,3,9,3,0]）

。我想提取出现

3

的索引，即

张量（[0,3,5]）

。有内置的功能吗？

有专门的功能：

   torch.where(my_tensor == the_number)

为此，有专门的解决方案：

   torch.where(my_tensor == the_number)

非零打印火炬张量的所有非零位置

非零打印火炬张量的所有非零位置