thrust中的容器、迭代器、算法(与STL类似)案例

东方耀

thrust中的容器、迭代器、算法(与STL类似)案例


thrust中预定义的函数对象 如何自定义thrust的函数对象

1. #include <thrust/device_vector.h>
   
2. #include <thrust/transform.h>
   
3. #include <thrust/sequence.h>
   
4. #include <thrust/copy.h>
   
5. #include <thrust/fill.h>
   
6. #include <thrust/replace.h>
   
7. #include <thrust/functional.h>
   
8. #include <iostream>
   
9. 
   
10. //thrust中的容器、迭代器、算法(与STL类似)案例
    
11. //thrust中预定义的函数对象 如何自定义thrust的函数对象
    
12. 
    
13. void print_thrust_vector(thrust::device_vector<int>& v){
    
14.     for(thrust::device_vector<int>::iterator it=v.begin();it!=v.end();it++){
    
15.         std::cout << "value=" << *it << std::endl;
    
16.     }
    
17.     for(int i=0; i<v.size();i++){
    
18.         //没有用迭代器 用了[]运算符取值  operator[]
    
19.         std::cout << "value_[]操作符=" << v[i] << std::endl;
    
20.     }
    
21. }
    
22. 
    
23. int main()
    
24. {
    
25.   // allocate three device_vectors with 10 elements
    
26.   thrust::device_vector<int> X(10);
    
27.   thrust::device_vector<int> Y(10);
    
28.   thrust::device_vector<int> Z(10);
    
29. 
    
30.   // initialize X to 0,1,2,3, ....
    
31.   thrust::sequence(X.begin(), X.end());
    
32.   std::cout << "X的初始值：" << std::endl;
    
33.   print_thrust_vector(X);
    
34. 
    
35.   // compute Y = -X   
    
36.   thrust::transform(X.begin(), X.end(), Y.begin(), thrust::negate<int>());
    
37.   std::cout << "Y的初始值：" << std::endl;
    
38.   print_thrust_vector(Y);
    
39. 
    
40.   // fill Z with twos
    
41.   thrust::fill(Z.begin(), Z.end(), 2);
    
42.   std::cout << "Z的初始值：" << std::endl;
    
43.   print_thrust_vector(Z);
    
44. 
    
45. 
    
46. 
    
47.   // compute Y = X mod 2(因为Z全是2)
    
48.   thrust::transform(X.begin(), X.end(), Z.begin(), Y.begin(), thrust::modulus<int>());
    
49. 
    
50.   std::cout << "Y的值(Y = X mod 2)：" << std::endl;
    
51.   print_thrust_vector(Y);
    
52. 
    
53. 
    
54.   // replace all the ones in Y with tens
    
55.   thrust::replace(Y.begin(), Y.end(), 1, 10);
    
56.   std::cout << "Y的值(1替换为10)：" << std::endl;
    
57.   // print Y 打印值的另外方法：
    
58.   thrust::copy(Y.begin(), Y.end(), std::ostream_iterator<int>(std::cout, "\n"));
    
59.    
    
60.   return 0;   
    
61. }
    

复制代码

zouqiqi

让天下人人学会人工智能！人工智能的前景一片大好！