Numpy Lua Torch等价于np.where（）？

我有一个

ByteTensor

，想在

1

的地方获取索引。在numpy，我可以做类似的事情

a = np.array([1,0,1,0,1])
return np.where(a)

它将返回

（数组（[0,2,4]），

。Torch中是否定义了此功能

（在我的特殊情况下，我想使用这些索引对几个不同的张量对象进行索引，但最好知道如何进行索引。）

您可以使用，例如：

如果您确实需要仅查找1-s，则可以链接逻辑运算符：

> a = torch.ByteTensor{1,2,1,6,1}
> a:eq(1):nonzero()

> a = torch.ByteTensor{1,2,1,6,1}
> a:eq(1):nonzero()