从Eigen::Tensor创建tensorflow::Tensor

Question

如何从Eigen::Tensor创建tensorflow::Tensor？我可以一个接一个地复制元素，但我希望有更好的方法。

Answer 1

没有公共api可以在不复制数据的情况下从Eigen::Tensor创建tensorflow::Tensor。但是，您可以创建tensorflow::Tensor，并使用以下apis将其解释为Eigen::TensorMap：tensorflow::Tensor tf_tensor(tensor_constructor_args); // For the general case: Eigen::TensorMap<type_params> eigen_tensor = tf_tensor.tensor<Type, NumDims>(); // shortcuts if you know the tensor is a matrix/vector/scalar Eigen::TensorMap<type_params> eigen_matrix = tf_tensor.matrix<Type>(); Eigen::TensorMap<type_params> eigen_vector = tf_tensor.vector<Type>(); Eigen::TensorMap<type_params> eigen_scalar = tf_tensor.scalar<Type>();

这将避免复制。此外，特征张量和张量映射共享相同的apis，因此您可以互换使用它们。