C++ Eigen : dynamic tensor

我想实现一个具有张量 vector 作为成员的C++类。张量的尺寸不是预先定义的，而是根据某些输入数据取值。此外，张量的等级可以不同。像这样的东西:

std::vector< TensorXd > myTensors;

但是，在Eigen中，动态张量没有这种TensorXd类型。

为了构造每个张量，我将读取一个数据std::vector<double> values的 vector ，该 vector 表示维度n x n x ... x n(r次)的张量。像这样的东西:

Tensor<double, r> tensor = TensorMap<double, r>(values.data(), std::vector<size_t>(r, n);
myTensors.push_back(tensor);

有可能这样做吗？

预先非常感谢您的帮助!

更新:

正如Yaroslav Bulatov指出的那样，Eigen不支持动态等级，因此必须明确写出支持的等级。在我的代码中:

#include <iostream>
#include <vector>
#include <Eigen/Dense>
#include <unsupported/Eigen/CXX11/Tensor>

typedef Eigen::Tensor< double , 3 > Tensor3d;
typedef Eigen::Tensor< double , 4 > Tensor4d;
typedef Eigen::Tensor< double , 5 > Tensor5d;
typedef Eigen::Tensor< double , 6 > Tensor6d;
typedef Eigen::Tensor< double , 7 > Tensor7d;
typedef Eigen::Tensor< double , 8 > Tensor8d;
typedef Eigen::Tensor< double , 9 > Tensor9d;
typedef Eigen::Tensor< double , 10 > Tensor10d;

class MyClass
{
private:
    Eigen::MatrixXd Potentials_1;
    std::vector<Eigen::MatrixXd> Potentials_2;
    std::vector< Tensor3d > Potentials_3;
    std::vector< Tensor4d > Potentials_4;
    std::vector< Tensor5d > Potentials_5;
    std::vector< Tensor6d > Potentials_6;
    std::vector< Tensor7d > Potentials_7;
    std::vector< Tensor8d > Potentials_8;
    std::vector< Tensor9d > Potentials_9;
    std::vector< Tensor10d > Potentials_10;

public:
    MyClass();
    void setPotentials_1(const Eigen::MatrixXd &_Potentials_1){ Potentials_1 = _Potentials_1; }
    void setPotentials_2(const std::vector<Eigen::MatrixXd> &_Potentials_2){ Potentials_2 = _Potentials_2; }
    void setPotentials_3(const std::vector<Tensor3d> &_Potentials_3){ Potentials_3 = _Potentials_3; }
    void setPotentials_4(const std::vector<Tensor4d> &_Potentials_4){ Potentials_4 = _Potentials_4; }
    void setPotentials_5(const std::vector<Tensor5d> &_Potentials_5){ Potentials_5 = _Potentials_5; }
    void setPotentials_6(const std::vector<Tensor6d> &_Potentials_6){ Potentials_6 = _Potentials_6; }
    void setPotentials_7(const std::vector<Tensor7d> &_Potentials_7){ Potentials_7 = _Potentials_7; }
    void setPotentials_8(const std::vector<Tensor8d> &_Potentials_8){ Potentials_8 = _Potentials_8; }
    void setPotentials_9(const std::vector<Tensor9d> &_Potentials_9){ Potentials_9 = _Potentials_9; }
    void setPotentials_10(const std::vector<Tensor10d> &_Potentials_10){ Potentials_10 = _Potentials_10; }
};

雅罗斯拉夫(Yaroslav)还建议使用宏有助于使代码重复无效。我不熟悉C++宏，因此任何帮助将不胜感激。

谢谢你的帮助!

最佳答案

您可以 checkout xtensor C++模板库，该库支持动态和静态维。

http://xtensor.readthedocs.io/en/latest/

xtensor具有与numpy非常相似的API，包括矢量化，广播，通用功能。这里有一个numpy到xtensor备忘单:http://xtensor.readthedocs.io/en/latest/numpy.html

最后，您可以通过单击https://github.com/QuantStack/xtensor/顶部的活页夹徽章，在C++ Jupyter笔记本中尝试使用它

xtensor还附带了科学计算主要语言(R，Julia，Python)的绑定(bind)。