使用 ArrayFire 将向量添加到矩阵的所有列

Question

使用 ArrayFire 将向量添加到矩阵的所有列的最佳方法是什么？

目前我正在使用gfor循环，但对于如此简单的任务来说，这似乎是错误的。

gfor(af::seq i, M.dims(1)) {
    M(af::span, i) += VECTOR;
}

有没有更好的方法？

Answer 1

您可以使用tile .由于您正在平铺单例维度 (VECTOR.dims(1( = 1(，因此这将作为 JIT 操作(在同一内核中(完成，而不是调用不同的内核。

M += af::tile(VECTOR, 1, M.dims(1));