擅长:python、mysql、java
<p>假设您唯一的问题是未定义的<code>Status</code>类型,复制和粘贴教程代码就可以了,除了这个之外,您需要在第一次使用<code>Status</code>之前将{<cd2>}移动到,或者完全限定它(如<code>return tensorflow::Status::OK()</code>)</p>
<p>例如,如果使用模板版本,<code>REGISTER_OP</code>部分可以如下所示:</p>
<pre><code>REGISTER_OP("ZeroOut")
.Attr("T: {float, int32}")
.Input("to_zero: T")
.Output("zeroed: T")
.SetShapeFn([](::tensorflow::shape_inference::InferenceContext* c) {
c->set_output(0, c->input(0));
return tensorflow::Status::OK();
});
</code></pre>