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

Question

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

目前我正在使用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));