相关疑难解决方法(0)

我有张量

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

和一个查询张量

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

有没有办法获得q像的索引

indexes = t.index(q) # get back [0, 3]

在 pytorch 中？

我有 2 个名为x和list 的张量，它们的定义如下：

x = torch.tensor(3)
list = torch.tensor([1,2,3,4,5])

现在我想从list 中获取元素x的索引。预期的输出是一个整数：

2