10#ifndef EIGEN_CXX11_TENSOR_TENSOR_INITIALIZER_H
11#define EIGEN_CXX11_TENSOR_TENSOR_INITIALIZER_H
13#if EIGEN_HAS_VARIADIC_TEMPLATES
15#include <initializer_list>
26template <
typename Derived,
int N>
28 typedef std::initializer_list<
29 typename Initializer<Derived, N - 1>::InitList> InitList;
31 static void run(TensorEvaluator<Derived, DefaultDevice>& tensor,
32 Eigen::array<
typename traits<Derived>::Index, traits<Derived>::NumDimensions>* indices,
33 const InitList& vals) {
35 for (
const auto& v : vals) {
36 (*indices)[traits<Derived>::NumDimensions - N] = i++;
37 Initializer<Derived, N - 1>::run(tensor, indices, v);
42template <
typename Derived>
43struct Initializer<Derived, 1> {
44 typedef std::initializer_list<typename traits<Derived>::Scalar> InitList;
46 static void run(TensorEvaluator<Derived, DefaultDevice>& tensor,
47 Eigen::array<
typename traits<Derived>::Index, traits<Derived>::NumDimensions>* indices,
48 const InitList& vals) {
51 for (
const auto& v : vals) {
52 (*indices)[traits<Derived>::NumDimensions - 1] = i++;
53 tensor.coeffRef(*indices) = v;
58template <
typename Derived>
59struct Initializer<Derived, 0> {
60 typedef typename traits<Derived>::Scalar InitList;
62 static void run(TensorEvaluator<Derived, DefaultDevice>& tensor,
63 Eigen::array<
typename traits<Derived>::Index, traits<Derived>::NumDimensions>*,
65 tensor.coeffRef(0) = v;
70template <
typename Derived,
int N>
71void initialize_tensor(TensorEvaluator<Derived, DefaultDevice>& tensor,
72 const typename Initializer<Derived, traits<Derived>::NumDimensions>::InitList& vals) {
73 Eigen::array<typename traits<Derived>::Index, traits<Derived>::NumDimensions> indices;
74 Initializer<Derived, traits<Derived>::NumDimensions>::run(tensor, &indices, vals);
Namespace containing all symbols from the Eigen library.