torch7:获取张量中元素的频率
torch7: Obtaining the Frequency of an Element in a Tensor
我正在寻找一些有效的方法(或方法的组合)来计算实例的数量(即获得频率)给定元素在任何一般 torch.Tensor
.
我所说的高效,指的不是搜索给定张量的每个元素的简单迭代方法,而是可以利用 GPU 提供的并行性的方法。
只是将 Alex 的评论表述为答案。我发现它对我自己的工作很有用。示例:
t = torch.Tensor({1, 2, 3, 1, 4, 2, 2, 2, 3, 0})
frequency = t:eq(2):sum() -- frequency of 2 in t
我正在寻找一些有效的方法(或方法的组合)来计算实例的数量(即获得频率)给定元素在任何一般 torch.Tensor
.
我所说的高效,指的不是搜索给定张量的每个元素的简单迭代方法,而是可以利用 GPU 提供的并行性的方法。
只是将 Alex 的评论表述为答案。我发现它对我自己的工作很有用。示例:
t = torch.Tensor({1, 2, 3, 1, 4, 2, 2, 2, 3, 0})
frequency = t:eq(2):sum() -- frequency of 2 in t