我正在寻找一些高效方式(或多种方式的组合)来计算任何一般情况下给定元素的实例数(即获得频率){ {1}}。
通过高效,我的意思是除了简单的迭代方法之外,搜索给定张量的每个元素以及可以利用GPU提供的并行性的东西。
答案 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