我有 2 个名为x和list 的张量,它们的定义如下:
x = torch.tensor(3) list = torch.tensor([1,2,3,4,5])
现在我想从list 中获取元素x的索引。预期的输出是一个整数:
2
我怎样才能以简单的方式做到这一点?
python torch pytorch tensor
python ×1
pytorch ×1
tensor ×1
torch ×1