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

2016-5-24 20:54:34
收藏：0
阅读：72
评论：1

我在寻找一些高效的方法（或者一些组合方法）来计算任何通用 torch.Tensor 中给定元素的实例数量（即获取 频率）。

通过高效，我的意思是除了通过遍历给定张量中的每个元素来搜索的平凡方法之外，还有一些可以利用 GPU 提供的并行性的方法。

我查看了 tensor.md 和 maths.md 提供的文档，但没有找到能够帮助我的东西。

点赞
用户4917583
用户4917583

将 Alex 的评论转化为答案。这对我的工作很有用。例如：

t = torch.Tensor({1, 2, 3, 1, 4, 2, 2, 2, 3, 0})
frequency = t:eq(2):sum() -- t 中数字 2 的频率
2016-05-25 22:18:46

作者:

用户2452617

用户2452617
推荐文章