求两个向量的内积inner_product(thrust版)

东方耀

求两个向量的内积inner_product(thrust版)

1. #include <thrust/inner_product.h>
   
2. #include <thrust/device_vector.h>
   
3. 
   
4. #include <iostream>
   
5. 
   
6. using namespace std;
   
7. 
   
8. //求两个向量的内积inner_product(thrust版)
   
9. int main(){
   
10.     thrust::device_vector<int> v1,v2(7);   //注意这个细节
    
11.     //thrust::sequence(v1.begin(), v1.end());
    
12.     for(int i=0;i<5;i++){
    
13.         v1.push_back(i+1);
    
14.     }
    
15.     v2[0] = 4;
    
16.     v2[1] = 3;
    
17.     v2[2] = 2;
    
18.     v2[3] = 1;
    
19.     v2[4] = 0;
    
20.     v2[5] = 5;
    
21.     v2[6] = 6;
    
22.     cout << "数组初始值v1：" << endl;
    
23.     thrust::copy(v1.begin(), v1.end(), std::ostream_iterator<int>(std::cout, "\n"));
    
24.     cout << "数组初始值v2：" << endl;
    
25.     thrust::copy(v2.begin(), v2.end(), std::ostream_iterator<int>(std::cout, "\n"));
    
26. 
    
27.     //int inner_d1_d2 = std::inner_product(d1.begin(),d1.end(),d2.begin(), 1);
    
28.     int inner_d1_d2 = thrust::inner_product(v1.begin(),v1.end(),v2.begin(), 1);
    
29.     //结果=1 + 1*4 + 2*3 + 3*2 + 4*1 + 5*0    = 21 对的
    
30.     cout<<"v1与v2的内积(1)="<<inner_d1_d2<<endl;
    
31. 
    
32.     //inner_d1_d2 = std::inner_product(d1.begin(),d1.end(),d2.begin(), 3,multiplies<int>(),plus<int>());
    
33.     ////应用指定的二元操作,使用第一个操作代替加而第二个操作代替乘
    
34.     inner_d1_d2 = thrust::inner_product(v1.begin(),v1.end(),v2.begin(), 3,thrust::multiplies<int>(),thrust::plus<int>());
    
35.     //结果=3 * (1+4) * (2+3) * (3+2) * (4+1) * (5+0)   = 9375 对的
    
36.     cout<<"v1与v2的内积(3)="<<inner_d1_d2<<endl;
    
37. 
    
38.    
    
39. 
    
40.     return 0;
    
41. }
    
42. 
    
43. /*
    
44. 数组初始值v1：
    
45. 1
    
46. 2
    
47. 3
    
48. 4
    
49. 5
    
50. 数组初始值v2：
    
51. 4
    
52. 3
    
53. 2
    
54. 1
    
55. 0
    
56. 5
    
57. 6
    
58. v1与v2的内积(1)=21
    
59. v1与v2的内积(3)=9375
    
60. 
    
61. */
    
62. 
    
63. 
    
64. 
    

复制代码