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

时间:2016-05-24 15:08:23

标签: lua torch

我正在寻找一些高效方式(或多种方式的组合)来计算任何一般情况下给定元素的实例数(即获得频率){ {1}}。

通过高效,我的意思是除了简单的迭代方法之外,搜索给定张量的每个元素以及可以利用GPU提供的并行性的东西。

我查看了 tensor.md maths.md 提供的文档,但无法找到可以帮助我的内容。

1 个答案:

答案 0 :(得分:1)

只是将亚历克斯的评论作为答案。我发现它对我自己的工作很有用。例如:

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