2016-11-08 2 views
0

how to add a new operation in TensorFlowに関するチュートリアルです。共有オブジェクトを作成してPythonでロードする方法について説明します。Tensorflowの新しい操作をチュートリアルとは異なる方法で追加する

しかし、コア/カーネルにあるいくつかの操作では、共有オブジェクトが生成されませんでした。代わりに、TFはdist-packages/tensorflow/python/opsgen_some_ops.pyのようなファイルを生成します。

新しい操作(たとえばzero_out)を追加する2番目の方法を使用したい場合は、何をすべきですか(チュートリアルに加えて)?

ありがとうございました。

答えて

2

コアオペレータは_pywrap_tensorflow.soです。あなたがコアに貢献することを計画しているなら、それは彼らが行くところです。しかし、一般的には、最初にcontribディレクトリにそれらを受け入れ、その後の段階でコアに移動します。私たちと調整を始める前に、github issueを提出するのが最善です。

寄付を計画していない場合は、user_opsパスが最適です。そうすれば、バージョンの独立性をある程度保つことができます。

+0

私は現在、TFの運用の実装を変更するプロジェクトを行っています。だから私はコアモジュールで作業します。 'core/BUILD'に関連する文字列が見つからなかったので、' _pywrap_tensorflow.so'にどのようにリンクしているのか分かりませんでした。私は別の場所を探すべきですか? – fois

+0

これは 'tensorflow/python:pywrap_tensorflow'の' tf_py_wrap_cc'ルールです。 – drpng

+0

ありがとうございます。私は 'grep'を使うと何とかこのファイルを見逃しました。 – fois

関連する問題