在屬性python中定義tensorflow操作

Question

我想註冊一個python函數及其梯度作爲tensorflow操作。 我發現了許多有用的例子，例如：

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

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

儘管如此我希望登記在操作屬性，並通過調用op.get_attr使用漸變定義這些屬性（「attr_name」） 。

這是可能的，而不會下降到C實施？ 願你舉個例子嗎？

Answer 1

不幸的是我不相信有可能在不使用C++實現的操作的情況下添加屬性。一個可能有幫助的功能是，您可以通過在開頭添加下劃線來定義'私人'屬性。我不確定這是否記錄良好或長期保證是什麼，但您可以嘗試設置'_my_attr_name'，您應該能夠稍後檢索它。