从最低/最大数据点检索多个阵列阵列

Question

我有一个数组，其中包含一些感人的值。例如：

0 0 1 0 0 0 0 0 0 0
0 1 1 1 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 2 2 2 0 0
0 0 0 0 0 0 0 2 0 0
0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0
0 0 3 0 0 0 0 0 0 0
0 0 3 0 0 0 0 0 0 0
0 0 3 0 0 0 0 0 0 0

由此，我创建了一组 af::arrays：minX、maxX、minY、maxY。这些定义了包围每个组的框。 所以对于这个例子：

minX would be: [1,5,2] // 1 for label(1), 5 for label(2) and 2 for label(3)
maxX would be: [3,7,2] // 3 for label(1), 7 for label(2) and 2 for label(3)
minY would be: [0,3,7] // 0 for label(1), 3 for label(2) and 7 for label(3)
maxY would be: [1,4,9] // 1 for label(1), 4 for label(2) and 9 for label(3)

如果你从每个数组中取出第 i 个元素，你就可以获得包含相应标签的盒子的左上/右下边界。

我想使用这些值从这个更大的数组中提取子数组。我的目标是将这些包含在框中的值放入一个平面列表中。在 GPU 内存中，我还使用最大/最小 X/Y 值计算了每个框需要多少个条目。因此，在此示例中，平面列表的结果应为：

result=[0 1 0 1 1 1 2 2 2 0 0 2 3 3 3]

其中前 6 个条目来自框中，

 ______
|0 1 0 |
|1 1 1 |
 ------

后 6 个条目来自框

 ______
|2 2 2 |
|0 0 2 |
 ------

最后三个条目来自框中，

 ___
| 3 |
| 3 |
| 3 |
 ---

我无法弄清楚如何使用驻留在 GPU 上的内存中的最小/最大值来索引此 af::array （并且不想将它们传输到 CPU）。我试图看看 gfor/seq 是否适合我，但似乎 af::seq 无法使用数组数据，并且我尝试使用 af::index 的所有内容也无法为我工作。

我能够更改表示最小值/最大值的方式（我可以存储左上/右下的索引），但我的主要目标是在 GPU 上高效地完成此操作，而无需在 GPU 和 CPU 之间来回移动数据。

如何使用 ArrayFire 有效地实现这一目标？

感谢您的帮助

Answer 1

到目前为止你是怎么到达那里的？您使用哪种语言？

我猜你可以将结果平铺到第三维来分别处理每个区域，并最终在 GPU 内存中得到最小/最大向量。