我可以在ArrayFire使用广播吗

Question

我能做这样的事情吗：

auto minEl = min(a);
a -= minEl;

当我这样做的时候，我会得到一个未知的af：：异常。现在，我正在做这个：

auto minEl = *min(a).host<float>();
a -= minEl;

当然，它会进行不必要的下载。

我借用numpy的术语"广播"，因为它在那里工作得很好：(

Answer 1

ArrayFire目前不支持广播。您必须手动平铺阵列以匹配所需的尺寸
auto minEl = min(a); a -= tile(minEl, a.dims(0));

此方法还避免了将标量复制到主机内存。