how to add a new operation in TensorFlowに関するチュートリアルです。共有オブジェクトを作成してPythonでロードする方法について説明します。Tensorflowの新しい操作をチュートリアルとは異なる方法で追加する
しかし、コア/カーネルにあるいくつかの操作では、共有オブジェクトが生成されませんでした。代わりに、TFはdist-packages/tensorflow/python/ops
にgen_some_ops.py
のようなファイルを生成します。
新しい操作(たとえばzero_out
)を追加する2番目の方法を使用したい場合は、何をすべきですか(チュートリアルに加えて)?
ありがとうございました。
私は現在、TFの運用の実装を変更するプロジェクトを行っています。だから私はコアモジュールで作業します。 'core/BUILD'に関連する文字列が見つからなかったので、' _pywrap_tensorflow.so'にどのようにリンクしているのか分かりませんでした。私は別の場所を探すべきですか? – fois
これは 'tensorflow/python:pywrap_tensorflow'の' tf_py_wrap_cc'ルールです。 – drpng
ありがとうございます。私は 'grep'を使うと何とかこのファイルを見逃しました。 – fois