cxx11_tensor_dimension.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 
16 
17 static void test_dynamic_size()
18 {
20 
21  VERIFY_IS_EQUAL((int)Eigen::internal::array_get<0>(dimensions), 2);
22  VERIFY_IS_EQUAL((int)Eigen::internal::array_get<1>(dimensions), 3);
23  VERIFY_IS_EQUAL((int)Eigen::internal::array_get<2>(dimensions), 7);
24  VERIFY_IS_EQUAL((int)dimensions.TotalSize(), 2*3*7);
25  VERIFY_IS_EQUAL((int)dimensions[0], 2);
26  VERIFY_IS_EQUAL((int)dimensions[1], 3);
27  VERIFY_IS_EQUAL((int)dimensions[2], 7);
28 }
29 
30 static void test_fixed_size()
31 {
33 
34  VERIFY_IS_EQUAL((int)Eigen::internal::array_get<0>(dimensions), 2);
35  VERIFY_IS_EQUAL((int)Eigen::internal::array_get<1>(dimensions), 3);
36  VERIFY_IS_EQUAL((int)Eigen::internal::array_get<2>(dimensions), 7);
37  VERIFY_IS_EQUAL((int)dimensions.TotalSize(), 2*3*7);
38 }
39 
40 static void test_match()
41 {
42  Eigen::DSizes<unsigned int, 3> dyn((unsigned int)2,(unsigned int)3,(unsigned int)7);
44  VERIFY_IS_EQUAL(Eigen::dimensions_match(dyn, stat), true);
45 
46  Eigen::DSizes<int, 3> dyn1(2,3,7);
47  Eigen::DSizes<int, 2> dyn2(2,3);
48  VERIFY_IS_EQUAL(Eigen::dimensions_match(dyn1, dyn2), false);
49 }
50 
51 static void test_rank_zero()
52 {
54  VERIFY_IS_EQUAL((int)scalar.TotalSize(), 1);
55  VERIFY_IS_EQUAL((int)scalar.rank(), 0);
56  VERIFY_IS_EQUAL((int)internal::array_prod(scalar), 1);
57 
59  VERIFY_IS_EQUAL((int)dscalar.TotalSize(), 1);
60  VERIFY_IS_EQUAL((int)dscalar.rank(), 0);
61 }
62 
64  Eigen::DSizes<int, 3> src0(1, 2, 3);
66  src1[0] = 4;
67  src1[1] = 5;
68  src1[2] = 6;
69 
70  Eigen::DSizes<long, 3> dst0(src0);
71  Eigen::DSizes<long, 3> dst1(src1);
72 
73  VERIFY_IS_EQUAL(dst0[0], 1L);
74  VERIFY_IS_EQUAL(dst0[1], 2L);
75  VERIFY_IS_EQUAL(dst0[2], 3L);
76  VERIFY_IS_EQUAL(dst1[0], 4L);
77  VERIFY_IS_EQUAL(dst1[1], 5L);
78  VERIFY_IS_EQUAL(dst1[2], 6L);
79 }
80 
81 EIGEN_DECLARE_TEST(cxx11_tensor_dimension)
82 {
88 }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE std::ptrdiff_t array_prod(const Sizes< Indices... > &)
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE std::ptrdiff_t rank() const
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE bool dimensions_match(Dims1 dims1, Dims2 dims2)
MatrixXd L
Definition: LLT_example.cpp:6
static void test_fixed_size()
#define VERIFY_IS_EQUAL(a, b)
Definition: main.h:386
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex TotalSize() const
static void test_index_type_promotion()
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index rank() const
static void test_match()
#define CALL_SUBTEST(FUNC)
Definition: main.h:399
EIGEN_DECLARE_TEST(cxx11_tensor_dimension)
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE std::ptrdiff_t TotalSize()
static void test_dynamic_size()
const std::vector< size_t > dimensions
mxArray * scalar(mxClassID classid)
Definition: matlab.h:82
static void test_rank_zero()
The tensor class.
Definition: Tensor.h:63


gtsam
Author(s):
autogenerated on Tue Jul 4 2023 02:34:07