cuda - 计算条件均值

Question

我是 cuda 编程的新手，并且有兴趣实现一种算法，该算法在连续编码时一次通过向量计算两个或多个均值。在 cuda 中执行此类操作的有效方案是什么？

有两个长度为 N 的向量，元素值和一个标识每个元素属于哪个子集的指示符值。

是否有一种有效的方法可以在一次通过中完成此操作，或者应该在 M 次通过中完成，其中 M 是要计算的均值的数量，并使用索引键向量作为每个子集的元素值？

Answer 1

您可以通过一次调用 thrust::reduce_by_key 传递数据来实现此目的.特别是，看看 "summary statistics" example ，它一次计算单个向量的多个统计属性。您可以将此方法概括为 reduce_by_key，它并行计算许多子向量的缩减。您的“指标值”将提供 reduce_by_key 用于确定每个元素属于哪个子向量的“键”。