这至少部分地是流压缩问题。Thrust提供流压缩算法，如copy_if。
Thrust确实有一个transform变体，它接受两个向量作为输入。你也可以用zip_iterator做类似的事情，作为任何推力算法的输入。
我认为最紧凑的方法是同时使用copy_if和zip迭代器和计数迭代器。有一个copy_if变体可以接受一个模板数组，它可以很好地满足我们这里的目的。
下面是一个示例：

# cat t61.cu
#include <thrust/copy.h>
#include <thrust/iterator/zip_iterator.h>
#include <thrust/iterator/counting_iterator.h>
#include <thrust/device_vector.h>
#include <thrust/host_vector.h>

using mt = int;  // data type
using it = size_t; // index type

struct my_functor
{
  template <typename T>
  __host__ __device__
  bool operator()(T t) { return (thrust::get<0>(t) > thrust::get<1>(t));}
};

int main(){

  mt a[] = {1,3,5,6,9};
  mt b[] = {2,1,4,7,8};
  it len = sizeof(a)/sizeof(a[0]);
  thrust::device_vector<mt> da(a, a+len);
  thrust::device_vector<mt> db(b, b+len);
  thrust::device_vector<it> dr(len);
  auto my_idx = thrust::counting_iterator<it>(0);
  auto my_zip = thrust::make_zip_iterator(thrust::make_tuple(da.begin(), db.begin()));
  it lr = thrust::copy_if(my_idx, my_idx+len, my_zip, dr.begin(), my_functor()) - dr.begin();
  thrust::host_vector<it> hr = dr;
  thrust::copy_n(hr.begin(), lr, std::ostream_iterator<it>(std::cout, ","));
  std::cout << std::endl;
}
# nvcc -o t61 t61.cu
# compute-sanitizer ./t61
========= COMPUTE-SANITIZER
1,2,4,
========= ERROR SUMMARY: 0 errors