cuda - 如何指定向量的范围？

Question

我创建了两个这样的向量

thrust::host_vector<int> A(count);
thrust::generate(A.begin(),A.end(),rand);

thrust::host_vector<int> B(count);
thrust::generate(B.begin(),B.end(),rand);

count 指定向量的大小是吗？我想用 0 100 个随机数填充这个向量。我怎样才能做到这一点？

Answer 1

构造函数中的 count 参数host_vector是向量的大小。请参阅host_vector文档。

根据generate推力文件

generate将调用的结果（gen一个不带参数的函数对象）分配给范围 [first,last) 中的每个元素。

因此，创建没有参数的随机函数

int rand_from_0_to_100_gen(void)
{
     return rand() % 100;
}

并在thrust::generate函数中使用

thrust::generate(A.begin(),A.end(),rand_from_0_to_100_gen);