torch7:获取张量中元素的频率

torch7: Obtaining the Frequency of an Element in a Tensor

我正在寻找一些有效的方法(或方法的组合)来计算实例的数量(即获得频率)给定元素在任何一般 torch.Tensor.

我所说的高效,指的不是搜索给定张量的每个元素的简单迭代方法,而是可以利用 GPU 提供的并行性的方法。

我查看了 tensor.md and maths.md 中提供的文档,但找不到可以帮助我的东西。

只是将 Alex 的评论表述为答案。我发现它对我自己的工作很有用。示例:

t = torch.Tensor({1, 2, 3, 1, 4, 2, 2, 2, 3, 0})
frequency = t:eq(2):sum() -- frequency of 2 in t