c++ - ArrayFire GFOR 循环意外错误

Question

我正在尝试使用 GFOR 结构，但我什至不能让它用于基本的东西。

我怀疑 GFOR 循环不适用于线性索引，对吗？

这里有一些不起作用的代码示例

af::array x(100, 200);
af::array y(100, 200);
af::array xx(100, 200);
gfor(af::seq i, (100 * 200)) {
    xx(i) = x(i) + y(i);
}

这个应该做一个简单的元素加法(我知道我可以做 xx = x+y，我只是想证明 gfor 没有按预期工作)。错误是函数 seqToDims 中的输入大小无效:203。

还有一个是比较简单的程序，像这样

af::array x(100, 200);
af::array y(100, 200);
gfor(af::seq i, (100 * 200)) {
    y(i) = x(i) + 1;
}

同样的错误。

我看过文档，我认为 gfor 不适用于线性索引，我需要以某种方式“逐行”或“逐列”工作，因为文档中的示例总是显示类似 A(span,i) 而不是 A(i) 的内容。

除此之外，我真正想做的是一些简单的事情:我有:数组图像(行，列)，数组 x_(8, 行*列),一个数组 ax(8,1)；和一个由 sum(x_(span,i)*ax) 找到的标量(它实际上是点积，但是 gfor 不支持 dot() 函数)，这个标量是使用 x_(大小为 8)的列和大小为 8 的 ax 计算的。

我要

e(i) = image(i) - sum<float>(ax * x_(span,i));

我似乎无法让它工作，因为即使是上面的简单 gfor 循环也不起作用，这是出乎意料的，我做错了什么？

谢谢!

Answer 1

您已经声明了二维数组，但您试图仅用一维来寻址它们。你不能有

array x(100, 200);
x(150) = 10;   // The second dimension is missing here.

您可以使用 gfor 创建一个序列，并在其中一个维度中使用该序列。您可以使用通常的方法来处理另一个维度，即“跨度”。

array x(100, 200); // 100 columns, 200 rows
gfor (seq i, 200) {   // will create a sequence from 0...199
   x(span, i) += 10;  // will take a slice of x: the entire column (span) of row 'i'
                      // and add the value 10 to it
}

注意 x 现在是用二维来处理的吗？第一个维度是 span，换句话说，就是那个维度的整个跨度。第二个维度是gfor中定义的序列。您当然可以交换尺寸:

array x(100, 200); // same as above
gfor (seq i, 100) {  // create  a sequence from 0..99  (not 199)
   x(i, span) += 20; // access x by rows (span) and the sequence i
}

从您的代码中我可以推断出您仍在线性/顺序思考，就像 C(++) 程序员一样。赠品是 seq i, (100*200) 您正在尝试遍历整个元素 (rows*cols)，但 arrayfire 使用 vector 算术。您不想访问单个元素。您想要访问 vector/切片/整行或整列并对它们执行某些操作。

对于您实际尝试完成的内容 (e(i) = image(i)...)，请查看 arrayfire 批处理函数。这可能会有所帮助。