连接（堆叠）Eigen::Tensors 以创建另一个张量答案

【问题标题】：Concatenate (stack up) Eigen::Tensors to create another tensor连接（堆叠）Eigen::Tensors 以创建另一个张量
【发布时间】：2019-07-27 10:26:36
【问题描述】：

我想将 3 个 2D 张量连接或叠加到 3D 张量。如何在 Eigen::Tensor 中做到这一点？

代码：

#include <iostream>
#include <CXX11/Tensor>

int main()
{
    Eigen::Tensor<float, 3> u(4, 4, 3);
    Eigen::Tensor<float, 2> a(4,4), b(4,4), c(4,4);
    a.setValues({{1, 2, 3, 4},     {5, 6, 7, 8},     {9, 10, 11, 12},  {13, 14, 15, 16}});
    b.setValues({{17, 18, 19, 20}, {21, 22, 23, 24}, {25, 26, 27, 28}, {29, 30, 31, 32}});
    c.setValues({{33, 34, 35, 36}, {37, 38, 39, 40}, {41, 42, 43, 44}, {45, 46, 47, 48}});
    u.concatenate(a, 0);
    u.concatenate(b, 0);
    u.concatenate(c, 0);

    std::cout<<u<<std::endl;
}

我得到的输出：

0 0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0

我做错了什么？我当然可以设置嵌套的 for 循环来实现我想要的，但我想知道库中是否有方法。此外，最好的解决方案是避免数据复制，并且可以将数据从源张量移动到目标张量，因为之后我将不再需要单独的 2D 张量。

【问题讨论】：

标签： c++11 eigen tensor eigen3

【解决方案1】：

concatenate 返回一个可以稍后计算的表达式。如果您想强制评估并分配给u，您可以执行以下操作：

u = a.concatenate(b, 0).eval().concatenate(c, 0).
        reshape(Eigen::Tensor<float, 3>::Dimensions(4, 4, 3));

【讨论】：