cxx11_tensor_assign.cpp
Go to the documentation of this file.
1 // This file is part of Eigen, a lightweight C++ template library
2 // for linear algebra.
3 //
4 // Copyright (C) 2014 Benoit Steiner <benoit.steiner.goog@gmail.com>
5 //
6 // This Source Code Form is subject to the terms of the Mozilla
7 // Public License v. 2.0. If a copy of the MPL was not distributed
8 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
9 
10 #include "main.h"
11 
12 #include <Eigen/CXX11/Tensor>
13 
14 using Eigen::Tensor;
15 using Eigen::RowMajor;
16 
17 static void test_1d()
18 {
21  vec1(0) = 4; vec2(0) = 0;
22  vec1(1) = 8; vec2(1) = 1;
23  vec1(2) = 15; vec2(2) = 2;
24  vec1(3) = 16; vec2(3) = 3;
25  vec1(4) = 23; vec2(4) = 4;
26  vec1(5) = 42; vec2(5) = 5;
27 
28  int col_major[6];
29  int row_major[6];
30  memset(col_major, 0, 6*sizeof(int));
31  memset(row_major, 0, 6*sizeof(int));
32  TensorMap<Tensor<int, 1> > vec3(col_major, 6);
34 
35  vec3 = vec1;
36  vec4 = vec2;
37 
38  VERIFY_IS_EQUAL(vec3(0), 4);
39  VERIFY_IS_EQUAL(vec3(1), 8);
40  VERIFY_IS_EQUAL(vec3(2), 15);
41  VERIFY_IS_EQUAL(vec3(3), 16);
42  VERIFY_IS_EQUAL(vec3(4), 23);
43  VERIFY_IS_EQUAL(vec3(5), 42);
44 
45  VERIFY_IS_EQUAL(vec4(0), 0);
46  VERIFY_IS_EQUAL(vec4(1), 1);
47  VERIFY_IS_EQUAL(vec4(2), 2);
48  VERIFY_IS_EQUAL(vec4(3), 3);
49  VERIFY_IS_EQUAL(vec4(4), 4);
50  VERIFY_IS_EQUAL(vec4(5), 5);
51 
52  vec1.setZero();
53  vec2.setZero();
54  vec1 = vec3;
55  vec2 = vec4;
56 
57  VERIFY_IS_EQUAL(vec1(0), 4);
58  VERIFY_IS_EQUAL(vec1(1), 8);
59  VERIFY_IS_EQUAL(vec1(2), 15);
60  VERIFY_IS_EQUAL(vec1(3), 16);
61  VERIFY_IS_EQUAL(vec1(4), 23);
62  VERIFY_IS_EQUAL(vec1(5), 42);
63 
64  VERIFY_IS_EQUAL(vec2(0), 0);
65  VERIFY_IS_EQUAL(vec2(1), 1);
66  VERIFY_IS_EQUAL(vec2(2), 2);
67  VERIFY_IS_EQUAL(vec2(3), 3);
68  VERIFY_IS_EQUAL(vec2(4), 4);
69  VERIFY_IS_EQUAL(vec2(5), 5);
70 }
71 
72 static void test_2d()
73 {
74  Tensor<int, 2> mat1(2,3);
75  Tensor<int, 2, RowMajor> mat2(2,3);
76 
77  mat1(0,0) = 0;
78  mat1(0,1) = 1;
79  mat1(0,2) = 2;
80  mat1(1,0) = 3;
81  mat1(1,1) = 4;
82  mat1(1,2) = 5;
83 
84  mat2(0,0) = 0;
85  mat2(0,1) = 1;
86  mat2(0,2) = 2;
87  mat2(1,0) = 3;
88  mat2(1,1) = 4;
89  mat2(1,2) = 5;
90 
91  int col_major[6];
92  int row_major[6];
93  memset(col_major, 0, 6*sizeof(int));
94  memset(row_major, 0, 6*sizeof(int));
95  TensorMap<Tensor<int, 2> > mat3(row_major, 2, 3);
96  TensorMap<Tensor<int, 2, RowMajor> > mat4(col_major, 2, 3);
97 
98  mat3 = mat1;
99  mat4 = mat2;
100 
101  VERIFY_IS_EQUAL(mat3(0,0), 0);
102  VERIFY_IS_EQUAL(mat3(0,1), 1);
103  VERIFY_IS_EQUAL(mat3(0,2), 2);
104  VERIFY_IS_EQUAL(mat3(1,0), 3);
105  VERIFY_IS_EQUAL(mat3(1,1), 4);
106  VERIFY_IS_EQUAL(mat3(1,2), 5);
107 
108  VERIFY_IS_EQUAL(mat4(0,0), 0);
109  VERIFY_IS_EQUAL(mat4(0,1), 1);
110  VERIFY_IS_EQUAL(mat4(0,2), 2);
111  VERIFY_IS_EQUAL(mat4(1,0), 3);
112  VERIFY_IS_EQUAL(mat4(1,1), 4);
113  VERIFY_IS_EQUAL(mat4(1,2), 5);
114 
115  mat1.setZero();
116  mat2.setZero();
117  mat1 = mat3;
118  mat2 = mat4;
119 
120  VERIFY_IS_EQUAL(mat1(0,0), 0);
121  VERIFY_IS_EQUAL(mat1(0,1), 1);
122  VERIFY_IS_EQUAL(mat1(0,2), 2);
123  VERIFY_IS_EQUAL(mat1(1,0), 3);
124  VERIFY_IS_EQUAL(mat1(1,1), 4);
125  VERIFY_IS_EQUAL(mat1(1,2), 5);
126 
127  VERIFY_IS_EQUAL(mat2(0,0), 0);
128  VERIFY_IS_EQUAL(mat2(0,1), 1);
129  VERIFY_IS_EQUAL(mat2(0,2), 2);
130  VERIFY_IS_EQUAL(mat2(1,0), 3);
131  VERIFY_IS_EQUAL(mat2(1,1), 4);
132  VERIFY_IS_EQUAL(mat2(1,2), 5);
133 }
134 
135 static void test_3d()
136 {
137  Tensor<int, 3> mat1(2,3,7);
138  Tensor<int, 3, RowMajor> mat2(2,3,7);
139 
140  int val = 0;
141  for (int i = 0; i < 2; ++i) {
142  for (int j = 0; j < 3; ++j) {
143  for (int k = 0; k < 7; ++k) {
144  mat1(i,j,k) = val;
145  mat2(i,j,k) = val;
146  val++;
147  }
148  }
149  }
150 
151  int col_major[2*3*7];
152  int row_major[2*3*7];
153  memset(col_major, 0, 2*3*7*sizeof(int));
154  memset(row_major, 0, 2*3*7*sizeof(int));
155  TensorMap<Tensor<int, 3> > mat3(col_major, 2, 3, 7);
156  TensorMap<Tensor<int, 3, RowMajor> > mat4(row_major, 2, 3, 7);
157 
158  mat3 = mat1;
159  mat4 = mat2;
160 
161  val = 0;
162  for (int i = 0; i < 2; ++i) {
163  for (int j = 0; j < 3; ++j) {
164  for (int k = 0; k < 7; ++k) {
165  VERIFY_IS_EQUAL(mat3(i,j,k), val);
166  VERIFY_IS_EQUAL(mat4(i,j,k), val);
167  val++;
168  }
169  }
170  }
171 
172  mat1.setZero();
173  mat2.setZero();
174  mat1 = mat3;
175  mat2 = mat4;
176 
177  val = 0;
178  for (int i = 0; i < 2; ++i) {
179  for (int j = 0; j < 3; ++j) {
180  for (int k = 0; k < 7; ++k) {
181  VERIFY_IS_EQUAL(mat1(i,j,k), val);
182  VERIFY_IS_EQUAL(mat2(i,j,k), val);
183  val++;
184  }
185  }
186  }
187 }
188 
189 static void test_same_type()
190 {
191  Tensor<int, 1> orig_tensor(5);
192  Tensor<int, 1> dest_tensor(5);
193  orig_tensor.setRandom();
194  dest_tensor.setRandom();
195  int* orig_data = orig_tensor.data();
196  int* dest_data = dest_tensor.data();
197  dest_tensor = orig_tensor;
198  VERIFY_IS_EQUAL(orig_tensor.data(), orig_data);
199  VERIFY_IS_EQUAL(dest_tensor.data(), dest_data);
200  for (int i = 0; i < 5; ++i) {
201  VERIFY_IS_EQUAL(dest_tensor(i), orig_tensor(i));
202  }
203 
204  TensorFixedSize<int, Sizes<5> > orig_array;
205  TensorFixedSize<int, Sizes<5> > dest_array;
206  orig_array.setRandom();
207  dest_array.setRandom();
208  orig_data = orig_array.data();
209  dest_data = dest_array.data();
210  dest_array = orig_array;
211  VERIFY_IS_EQUAL(orig_array.data(), orig_data);
212  VERIFY_IS_EQUAL(dest_array.data(), dest_data);
213  for (int i = 0; i < 5; ++i) {
214  VERIFY_IS_EQUAL(dest_array(i), orig_array(i));
215  }
216 
217  int orig[5] = {1, 2, 3, 4, 5};
218  int dest[5] = {6, 7, 8, 9, 10};
219  TensorMap<Tensor<int, 1> > orig_map(orig, 5);
220  TensorMap<Tensor<int, 1> > dest_map(dest, 5);
221  orig_data = orig_map.data();
222  dest_data = dest_map.data();
223  dest_map = orig_map;
224  VERIFY_IS_EQUAL(orig_map.data(), orig_data);
225  VERIFY_IS_EQUAL(dest_map.data(), dest_data);
226  for (int i = 0; i < 5; ++i) {
227  VERIFY_IS_EQUAL(dest[i], i+1);
228  }
229 }
230 
231 static void test_auto_resize()
232 {
233  Tensor<int, 1> tensor1;
234  Tensor<int, 1> tensor2(3);
235  Tensor<int, 1> tensor3(5);
236  Tensor<int, 1> tensor4(7);
237 
238  Tensor<int, 1> new_tensor(5);
239  new_tensor.setRandom();
240 
241  tensor1 = tensor2 = tensor3 = tensor4 = new_tensor;
242 
243  VERIFY_IS_EQUAL(tensor1.dimension(0), new_tensor.dimension(0));
244  VERIFY_IS_EQUAL(tensor2.dimension(0), new_tensor.dimension(0));
245  VERIFY_IS_EQUAL(tensor3.dimension(0), new_tensor.dimension(0));
246  VERIFY_IS_EQUAL(tensor4.dimension(0), new_tensor.dimension(0));
247  for (int i = 0; i < new_tensor.dimension(0); ++i) {
248  VERIFY_IS_EQUAL(tensor1(i), new_tensor(i));
249  VERIFY_IS_EQUAL(tensor2(i), new_tensor(i));
250  VERIFY_IS_EQUAL(tensor3(i), new_tensor(i));
251  VERIFY_IS_EQUAL(tensor4(i), new_tensor(i));
252  }
253 }
254 
255 
256 static void test_compound_assign()
257 {
258  Tensor<int, 1> start_tensor(10);
259  Tensor<int, 1> offset_tensor(10);
260  start_tensor.setRandom();
261  offset_tensor.setRandom();
262 
263  Tensor<int, 1> tensor = start_tensor;
264  tensor += offset_tensor;
265  for (int i = 0; i < 10; ++i) {
266  VERIFY_IS_EQUAL(tensor(i), start_tensor(i) + offset_tensor(i));
267  }
268 
269  tensor = start_tensor;
270  tensor -= offset_tensor;
271  for (int i = 0; i < 10; ++i) {
272  VERIFY_IS_EQUAL(tensor(i), start_tensor(i) - offset_tensor(i));
273  }
274 
275  tensor = start_tensor;
276  tensor *= offset_tensor;
277  for (int i = 0; i < 10; ++i) {
278  VERIFY_IS_EQUAL(tensor(i), start_tensor(i) * offset_tensor(i));
279  }
280 
281  tensor = start_tensor;
282  tensor /= offset_tensor;
283  for (int i = 0; i < 10; ++i) {
284  VERIFY_IS_EQUAL(tensor(i), start_tensor(i) / offset_tensor(i));
285  }
286 }
287 
289 #if EIGEN_HAS_VARIADIC_TEMPLATES
290  Tensor<int, 1> a(3);
291  a.setValues({0, 1, 2});
292  VERIFY_IS_EQUAL(a(0), 0);
293  VERIFY_IS_EQUAL(a(1), 1);
294  VERIFY_IS_EQUAL(a(2), 2);
295 
296  // It fills the top-left slice.
297  a.setValues({10, 20});
298  VERIFY_IS_EQUAL(a(0), 10);
299  VERIFY_IS_EQUAL(a(1), 20);
300  VERIFY_IS_EQUAL(a(2), 2);
301 
302  // Chaining.
303  Tensor<int, 1> a2(3);
304  a2 = a.setValues({100, 200, 300});
305  VERIFY_IS_EQUAL(a(0), 100);
306  VERIFY_IS_EQUAL(a(1), 200);
307  VERIFY_IS_EQUAL(a(2), 300);
308  VERIFY_IS_EQUAL(a2(0), 100);
309  VERIFY_IS_EQUAL(a2(1), 200);
310  VERIFY_IS_EQUAL(a2(2), 300);
311 
312  Tensor<int, 2> b(2, 3);
313  b.setValues({{0, 1, 2}, {3, 4, 5}});
314  VERIFY_IS_EQUAL(b(0, 0), 0);
315  VERIFY_IS_EQUAL(b(0, 1), 1);
316  VERIFY_IS_EQUAL(b(0, 2), 2);
317  VERIFY_IS_EQUAL(b(1, 0), 3);
318  VERIFY_IS_EQUAL(b(1, 1), 4);
319  VERIFY_IS_EQUAL(b(1, 2), 5);
320 
321  // It fills the top-left slice.
322  b.setValues({{10, 20}, {30}});
323  VERIFY_IS_EQUAL(b(0, 0), 10);
324  VERIFY_IS_EQUAL(b(0, 1), 20);
325  VERIFY_IS_EQUAL(b(0, 2), 2);
326  VERIFY_IS_EQUAL(b(1, 0), 30);
327  VERIFY_IS_EQUAL(b(1, 1), 4);
328  VERIFY_IS_EQUAL(b(1, 2), 5);
329 
330  Eigen::Tensor<int, 3> c(3, 2, 4);
331  c.setValues({{{0, 1, 2, 3}, {4, 5, 6, 7}},
332  {{10, 11, 12, 13}, {14, 15, 16, 17}},
333  {{20, 21, 22, 23}, {24, 25, 26, 27}}});
334  VERIFY_IS_EQUAL(c(0, 0, 0), 0);
335  VERIFY_IS_EQUAL(c(0, 0, 1), 1);
336  VERIFY_IS_EQUAL(c(0, 0, 2), 2);
337  VERIFY_IS_EQUAL(c(0, 0, 3), 3);
338  VERIFY_IS_EQUAL(c(0, 1, 0), 4);
339  VERIFY_IS_EQUAL(c(0, 1, 1), 5);
340  VERIFY_IS_EQUAL(c(0, 1, 2), 6);
341  VERIFY_IS_EQUAL(c(0, 1, 3), 7);
342  VERIFY_IS_EQUAL(c(1, 0, 0), 10);
343  VERIFY_IS_EQUAL(c(1, 0, 1), 11);
344  VERIFY_IS_EQUAL(c(1, 0, 2), 12);
345  VERIFY_IS_EQUAL(c(1, 0, 3), 13);
346  VERIFY_IS_EQUAL(c(1, 1, 0), 14);
347  VERIFY_IS_EQUAL(c(1, 1, 1), 15);
348  VERIFY_IS_EQUAL(c(1, 1, 2), 16);
349  VERIFY_IS_EQUAL(c(1, 1, 3), 17);
350  VERIFY_IS_EQUAL(c(2, 0, 0), 20);
351  VERIFY_IS_EQUAL(c(2, 0, 1), 21);
352  VERIFY_IS_EQUAL(c(2, 0, 2), 22);
353  VERIFY_IS_EQUAL(c(2, 0, 3), 23);
354  VERIFY_IS_EQUAL(c(2, 1, 0), 24);
355  VERIFY_IS_EQUAL(c(2, 1, 1), 25);
356  VERIFY_IS_EQUAL(c(2, 1, 2), 26);
357  VERIFY_IS_EQUAL(c(2, 1, 3), 27);
358 #endif // EIGEN_HAS_VARIADIC_TEMPLATES
359 }
360 
362 {
370 }
static void test_2d()
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index dimension(std::size_t n) const
Definition: Tensor.h:101
Scalar * b
Definition: benchVecAdd.cpp:17
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar * data()
static void test_same_type()
Scalar Scalar * c
Definition: benchVecAdd.cpp:17
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor< Scalar_, NumIndices_, Options_, IndexType_ > & setRandom()
Definition: TensorBase.h:850
MatrixXd mat1(size, size)
Array33i a
#define VERIFY_IS_EQUAL(a, b)
Definition: main.h:331
A tensor expression mapping an existing array of data.
void test_cxx11_tensor_assign()
static SO4::VectorN2 vec4(const Matrix4 &Q)
Definition: SO4.cpp:140
static void test_1d()
static void test_std_initializers_tensor()
The fixed sized version of the tensor class.
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar * data()
Definition: Tensor.h:104
RowVectorXd vec1(3)
#define CALL_SUBTEST(FUNC)
Definition: main.h:342
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PointerType data()
Definition: TensorMap.h:119
static Vector9 vec3(const Matrix3 &R)
Definition: SO3.cpp:307
static void test_auto_resize()
static void test_compound_assign()
static void test_3d()
std::ptrdiff_t j
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor< Scalar_, NumIndices_, Options_, IndexType_ > & setZero()
Definition: TensorBase.h:842
The tensor class.
Definition: Tensor.h:63


gtsam
Author(s):
autogenerated on Sat May 8 2021 02:41:55