opencl - 使用 PyOpenCL 的 argmin/argmax

Question

我将如何使用 PyOpenCL 编写 argmin 或 argmax？我想我需要计算每个工作组的 argmin/min，然后使用后续调用来减少这些。

Answer 1

调整它以收集最小值及其位置，而不仅仅是最小值。