Pytorch Tensor如何获得特定值的索引
python
pytorch
8
0

在python list中,我们可以使用list.index(somevalue) 。 pytorch如何做到这一点?
例如:

    a=[1,2,3]
    print(a.index(2))

然后,将输出1 。 pytorch张量如何在不将其转换为python列表的情况下做到这一点?

参考资料:
Stack Overflow
收藏
评论
共 1 个回答
高赞 时间 活跃

我认为没有从list.index()到pytorch函数的直接转换。但是,可以使用tensor==number然后使用nonzero()函数来获得类似的结果。例如:

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

这段代码返回

1个

[1x1大小的Torch.LongTensor]

收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号