到目前为止，在Thrust中还没有一个名为reduce_if的函数，这将是您正在搜索的。对于给定的函数，有多种方法可以做到这一点，哪种方法最适合你的问题可能取决于掩码中true s与false s的比例以及它们的分布方式。
也就是说，实现这一点的规范方法是将transform_reduce与zip_iterator一起使用：

#include <thrust/device_vector.h>
#include <thrust/functional.h>
#include <thrust/transform_reduce.h>
#include <thrust/iterator/zip_iterator.h>
#include <thrust/zip_function.h>

int reduce_if(thrust::device_vector<int> const &data,
              thrust::device_vector<bool> const &mask) {
  return thrust::transform_reduce(
      thrust::make_zip_iterator(thrust::make_tuple(
          data.cbegin(), mask.cbegin())),
      thrust::make_zip_iterator(thrust::make_tuple(
          data.cend(), mask.cend())),
      thrust::make_zip_function(
          [] __host__ __device__ (int value, bool flag){
            return flag ? value : 0;
          }),
      0,
      thrust::plus<int>{});
}

要编译这个，nvcc需要--extended-lambda标志。