在Tensorflow的C ++ API中，如何使用Eigen Tensor设置Tensorflow Tensor？

Question

所以我想说我有一个4D Eigen :: Tensor T.

同样地，我也有一个4D Tensorflow :: Tensor X，其形状与T相同

  int size = T.dimension(0);
  int rows = T.dimension(1);
  int cols = T.dimension(2);
  int channels = T.dimension(3);

  TensorShape TS;
  TS.AddDim(size);
  TS.AddDim(rows);
  TS.AddDim(cols);
  TS.AddDim(size);

  Tensor x( DT_FLOAT, TS);

现在我想把数据放在T中的x中。

所以我尝试做：

  x.matrix<float>()() = T;

但是当我做的时候，编译器会对我大喊：

无法将'Eigen :: Tensor'转换为'Eigen :: TensorMap，16> :: Scalar {aka float}'

当我尝试将T转换为TensorMap时，我得到更多错误。

我在这里错过了什么？

Answer 1

我认为问题是你使用'.matrix'，它只返回2d矩阵。

template <typename T>
  typename TTypes<T>::Matrix matrix() {
    return tensor<T, 2>();
  }

你应该使用'.tensor'，它可以返回Nd矩阵。

template <typename T, size_t NDIMS>
typename TTypes<T, NDIMS>::Tensor Tensor::tensor() {
  CheckTypeAndIsAligned(DataTypeToEnum<T>::v());
  return typename TTypes<T, NDIMS>::Tensor(base<T>(),
                                           shape().AsEigenDSizes<NDIMS>());
}