我可以在ArrayFire中使用广播吗？

Question

我可以做这样的事情吗？

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

？

当我这样做时，我得到了一个未知的af :: exception。现在，我正在这样做：

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

但是，当然，它不需要下载。

我从numpy借用了“广播”一词，因为它在这里非常有用：）

Answer 1

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

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