Tensorflow中的自定义新操作会导致异常';未分配要释放的指针';
我正在学习如何向tensorflow following添加操作,但遇到了一些麻烦 我从Github克隆了源代码,并计划将Tensorflow中的自定义新操作会导致异常';未分配要释放的指针';,tensorflow,Tensorflow,我正在学习如何向tensorflow following添加操作,但遇到了一些麻烦 我从Github克隆了源代码,并计划将tensorflow/core/user\u ops/Fact.cc中的Fact操作添加到框架中。我的做法如下: 在tensorflow/core/user_ops/中创建一个BUILD文件,内容如下: 在克隆项目的根目录中运行以下命令: bazel build-c opt-s//tensorflow/core/user_ops:fact.so 这确实会在bazel-bin/
tensorflow/core/user\u ops/Fact.cc
中的Fact
操作添加到框架中。我的做法如下:
tensorflow/core/user_ops/
中创建一个BUILD
文件,内容如下:bazel-bin/tensorflow/core/user_-ops/
事实复制到我的Python项目中,该项目位于另一个不相关的位置
有人能告诉我第一种方法失败的原因吗?您能告诉我您机器上的gcc版本吗?您能告诉我您机器上的gcc版本吗?
load("//tensorflow:tensorflow.bzl", "tf_custom_op_library")
tf_custom_op_library(
name = "fact.so",
srcs = ["fact.cc"]
)
import tensorflow as tf
fact_module = tf.load_op_library('fact.so')
with tf.Session(''):
print(fact_module.fact2().eval())