2

我有扩展推力矢量的特殊要求。假设我有键向量 K、值向量 V 和扩展因子向量 E，它对应于键向量。我想将与某个键对应的值复制（扩展因子）倍。我查看了几个 Thrust::expand 示例，但它似乎不适用于我的特定用途。通过thrust::reduce_by_key 为结果数组分配空间很容易，但我不知道如何实际扩展我的向量。

例如：

key is   [0,0,0,1,2,2,2,2,4]
value is [1,2,3,5,6,7,8,4,7]
key 0 has values [1,2,3] 
key 1 has value  [5]
key 2 has values [6,7,8,4]
key 4 has value  [7]
(This is not code but the website won't let me submit unless I indent these statements)

扩展因子数组：

Expansion factor: [2,3,1,1,3]
desired result array: [1,2,3,1,2,3,5,5,5,6,7,8,4,7,7,7]
1,2,3   are the values of key[0], expanded 2 times according to E[0]
5       is the value of key[1], expanded 3 times according to E[1]
6,7,8,4 are the values of key[2], expanded 1 times according to E[2]
[none]  is the value of key[3], expanded 1 times according to E[3]
7       is the value of key[4], expanded 3 times according to E[4]

有没有一种有效的方法来做到这一点？提前致谢。

4

1 回答 1

2

原发帖人报告了这个问题的解决方案如下：

  1. 获取三个辅助数组，它们是：每个键的开始位置，每个键的元素数，以及每个键的元素数（扩展后）的独占结果。
  2. 制作专属扫描结果数组的副本，并使用thrust::expand.
  3. 通过扩展数组使用计数迭代器，每个的起始位置都是key[(iterator - exclusive scan result)%number of elements]当前迭代器的结果

此社区 wiki 条目是从评论中添加的，以将问题从未回答列表中删除。

于 2014-05-28T08:54:00.767 回答