eigen - 如何在 Eigen 中转置张量

Question

我试图获得两个张量的矩阵乘积，其中一个张量应该在相乘之前转置( At*B )。

到目前为止，我在 eigen documentation 中发现了什么是没有任何转置和两个矩阵都转置的矩阵乘积。

我正在寻找一种方法来直接收缩两个张量，其中一个张量被转置，或者通过在收缩之前转置一个张量。

Answer 1

我想通了，转置效果可以使用 shuffle 方法来完成。

Eigen::Tensor<int, 2> m(3, 5);
m.setValues(
{
    {1, 2, 3, 4, 5},
    {6, 7, 8, 9, 10},
    {11, 12, 13, 14, 15}
});

Eigen::array<int, 2> shuffling({1, 0});

Eigen::Tensor<int, 2> transposed = m.shuffle(shuffling);
Eigen::Tensor<int, 2> original = transposed.shuffle(shuffling);