我应该如何从 Eigen::Tensor 创建一个 tensorflow::Tensor?我可以一个接一个地复制元素,但我希望有更好的方法。
最佳答案
没有公共(public) api 可以在不复制数据的情况下从 Eigen::Tensor 创建 tensorflow::Tensor。但是,您可以使用以下 API 创建一个 tensorflow::Tensor 并将其解释为 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>();
这将避免复制。此外, Eigen 张量和张量图共享相同的 API,因此您可以互换使用它们。
关于c++ - 从 Eigen::Tensor 创建 tensorflow::Tensor,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50744193/