小编Nic*_*ild的帖子

Pytorch Tensor 如何获取元素的索引?

我有 2 个名为xlist 的张量,它们的定义如下:

x = torch.tensor(3)
list = torch.tensor([1,2,3,4,5])
Run Code Online (Sandbox Code Playgroud)

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

2
Run Code Online (Sandbox Code Playgroud)

我怎样才能以简单的方式做到这一点?

python torch pytorch tensor

0
推荐指数
1
解决办法
4952
查看次数

标签 统计

python ×1

pytorch ×1

tensor ×1

torch ×1