属性を持つPythonのテンソルフロー演算を定義する

Question

私は、テンソルフロー演算としてPython関数とその勾配を登録しようとしています。 私は多くの有用な例などを発見した：

Write Custom Python-Based Gradient Function for an Operation? (without C++ Implementation)

https://programtalk.com/python-examples/tensorflow.python.framework.function.Defun/

それでも私は運転中の属性を登録したいと（「ATTR_NAME」）op.get_attr呼び出すことによって、勾配定義にこれらの属性を使用します。

これはC実装に移行することなく可能ですか？ 私に例を挙げてもらえますか？

Answer 1

残念ながら、私は操作のC++実装を使用せずに属性を追加することは可能ではないと思います。ただし、最初にアンダースコアを付けることで、「プライベート」属性を定義できるという点で役立ちます。私はこれが十分に文書化されているのか、長期的な保証があるのか​​は分かりませんが、 '_my_attr_name'を設定しようとすると後でそれを取得できるはずです。