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

Question

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

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

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

有没有更好的办法？

score 1 · Accepted Answer

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

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

1 回答 1