c++ - 在张量流中使用 eigen::matrix 乘法的最佳方法是什么？将张量转换为矩阵，然后将其转换回来？

Question

我想在 tensorflow 的 c++ 中添加一个新的操作。我想将矩阵乘法应用于张量。我将张量转换为 Eigen::matrix 以获得产品，然后将其转换回张量。

EigenTensor<float> Mh_chip = Tmh.chip(i, 0);
EigenTensor<float> h_chip = Th.chip(i, 0);
auto Mh = EigenMatrixMap<float>(Mh_chip.data(), Mh_chip.dimension(0), Mh_chip.dimension(1));
auto h = EigenMatrixMap<float>(h_chip.data(), h_chip.dimension(0), h_chip.dimension(1));
EigenMatrix<float> h_p = Mh * h;
Th_p.chip(i, 0) = EigenTensorMap<float>(h_p.data(), h_chip.dimensions());

写这样的代码很痛苦。有什么建议吗？