arrayfire - 数组条件过滤 (ArrayFire)

Question

我有两个数组 A（2D 图像）和 B（1d 强度值），并尝试制作第三个（2D）数组，其大小为 A，但仅包含 B 具有的 A 值。在不将数据移动到主机的情况下，正确的方法是什么？PS：试图找到一种方法来处理函数“区域”的结果（屏蔽某些区域并保留其他区域）。谢谢。