2016-05-24 21 views
0

我尋找一些高效方式(或方法的組合)中的任何一般計數給定元素的實例的數目(即獲得頻率torch.Tensortorch7:獲得元件的頻率在一個張量

通過高效,我的意思是除了平凡的迭代方法以外,搜索給定張量的每個元素以及可以利用GPU提供的並行性的東西。

我已經看過在tensor.mdmaths.md提供的文檔,但無法找到的東西,可以幫助我。

+1

'T:eq(x):sum()' –

回答

1

只是將Alex的評論作爲答案。我發現它對我自己的工作很有用。例如:

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