python - tensorflow::Tensor 到 python Tensor 或 numpy.nd_array

Question

我有一个 C++ tensorflow::Tensor我试图在 Python 中访问其内容。

我已经查看了 Tensorflow C++ Documentation寻找可以转换 tensorflow::Tensor 的函数任何明智的PyObject (现在不管是 tf.Tensor 还是 numpy.nd_array )。

在查看了 Tensorflow 代码之后，我发现了以下线索:

有一个方法TF_Tensor_to_PyObject(TF_Tensor* Tensor, PyObject** out_array)在 tensorflow/python/client/tf_session_helper.cc 中定义.然而，这是在一个隐藏的 Bazel 包中定义的，而且是在一个匿名的 C++ 命名空间中定义的。修改 Tensorflow 本身(我必须​​修改 Bazel BUILD 文件、.h 和 .cc 文件)并编译我自己的 Tensorflow 以使用此方法似乎不直观。

此方法的另一个问题是 TF_Tensor != tensorflow::Tensor . TF_Tensor在 Tensorflow 的 C-Api 中定义，并且转换也不打算供公众使用(如:在此包之外)。

有人知道更好的方法吗？是否有 tensorflow::Tensor 的现有实现到 PyObject我在搜索时没有找到？

Answer 1

py_func 操作的实现有 code to convert Tensor to PyObject .