mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
3625734bc8
Created the TensorDimensionList class to encode the list of all the dimensions of a tensor of rank n. This could be done using TensorIndexList, however TensorIndexList require cxx11 which isn't yet supported as widely as we'd like.
37 lines
1017 B
C++
37 lines
1017 B
C++
// This file is part of Eigen, a lightweight C++ template library
|
|
// for linear algebra.
|
|
//
|
|
// Copyright (C) 2015 Benoit Steiner <benoit.steiner.goog@gmail.com>
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla
|
|
// Public License v. 2.0. If a copy of the MPL was not distributed
|
|
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
#ifndef EIGEN_CXX11_TENSOR_TENSOR_META_H
|
|
#define EIGEN_CXX11_TENSOR_TENSOR_META_H
|
|
|
|
namespace Eigen {
|
|
|
|
template<bool cond> struct Cond {};
|
|
|
|
template<typename T1, typename T2> EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
|
|
const T1& choose(Cond<true>, const T1& first, const T2&) {
|
|
return first;
|
|
}
|
|
|
|
template<typename T1, typename T2> EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
|
|
const T2& choose(Cond<false>, const T1&, const T2& second) {
|
|
return second;
|
|
}
|
|
|
|
template <size_t n> struct max_n_1 {
|
|
static const size_t size = n;
|
|
};
|
|
template <> struct max_n_1<0> {
|
|
static const size_t size = 1;
|
|
};
|
|
|
|
} // namespace Eigen
|
|
|
|
#endif // EIGEN_CXX11_TENSOR_TENSOR_META_H
|