Definieren von Tensorflow-Operationen in Python mit Attributen

Question

Ich versuche, eine Python-Funktion und ihren Gradienten als eine Tensorflow-Operation zu registrieren. ich viele nützliche Beispiele zum Beispiel gefunden:

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

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

Trotzdem würde Ich mag Attribute im Betrieb registrieren und diese Attribute in den Gradienten-Definition verwenden, indem Aufruf op.get_attr (‚attr_name‘) .

Ist dies möglich ohne auf C-Implementierung zu gehen? Können Sie mir ein Beispiel geben?

Answer 1

Leider glaube ich nicht, dass es möglich ist, Attribute hinzuzufügen, ohne eine C++ - Implementierung der Operation zu verwenden. Eine Funktion, die helfen kann, ist, dass Sie "private" Attribute definieren können, indem Sie dem Start einen Unterstrich voranstellen. Ich bin nicht sicher, ob dies gut dokumentiert ist oder was die langfristigen Garantien sind, aber Sie können versuchen, '_my_attr_name' zu setzen und Sie sollten es später abrufen können.