不同等级的特征张量向量

Question

我的问题：是否可以创建具有不同等级的特征张量向量？

我的目标是能够创建一个包含对象的向量，这些对象的类型取值在Eigen::MatrixXd, Tensor3d, Tensor4d,..., Tensor10d（定义如下）之间。对象可以有不同的类型。

提前感谢您的帮助！

#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:
    std::vector< TensorXd > Tensors;
public:
    MyClass();
};

Answer 1

由于您将此问题标记为[tensorflow]，我假设您将Eigen::Tensor 用作TensorFlow 程序的一部分。 answer Aziuth pointed in their comment 建议使用包装类来保存张量，幸运的是 TensorFlow 带有自己的包装类：tensorflow::Tensor。

您可以创建 std::vector<tensorflow::Tensor>，使用 TensorFlow 的分配器为不同的张量分配内存，并使用 tensorflow::Tensor::tensor() 方法访问包装好的 Eigen::Tensor 对象。