Halide：如何显式安排匿名归约函数，例如sum？

Question

这是用Halide编写的卷积函数： 1 4 7 11 2 3 8 3 7 9 10

html如下所示：

我有点希望可以安排总和，例如： output(x, y, z, n) = sum(Input(x * stride + r.x, y * stride + r.y, r.z, n) * Kernel(r.x, r.y, r.z, z));

所以总和可以不止一个元素。

我该如何使用Halide？