torch 中的 ByteTensor 逻辑取反

是否有一种方法可以在 Torch 中对 ByteTensor 进行逻辑否定(即进行 NOT 操作):

th> a = torch.rand(4)

th> a
 0.5786
 0.5271
 0.0090
 0.8859
[torch.DoubleTensor of size 4]

th> b = a:le(0.5)

th> a[b]
0.001 *
 9.0080
[torch.DoubleTensor of size 1]

th> -- 如何选择 a 的所有“其他”元素?
th> -- a[~b]、a[!b] 或 a[b:neg()] 等方法都不起作用。
点赞
用户1688185
用户1688185

通过 torch.ne(不等于)的方法:

a[b:ne(1)]

编辑

或者等价于通过 torch.eq(等于)的方法:

a[b:eq(0)]
2016-04-04 15:44:54