我有一个 ByteTensor
并想获取有 1
的索引。在 numpy 中,我可以做类似的事情
a = np.array([1,0,1,0,1])
return np.where(a)
这将返回 (array([0, 2, 4]),)
。这个功能是在 Torch 中定义的吗?
(在我的特定情况下,我想使用这些索引来索引几个不同的 Tensor 对象,但最好知道一般情况下如何执行此操作。)
最佳答案
您可以使用 torch.nonzero ,例如:
> a = torch.ByteTensor{1,0,1,0,1}
> print(torch.nonzero(a))
1
3
5
[torch.LongTensor of size 3x1]
如果你真的需要找到 1-s only 你可以链接一个逻辑运算符:
> a = torch.ByteTensor{1,2,1,6,1}
> a:eq(1):nonzero()
关于numpy - Lua Torch 相当于 np.where()?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35881370/