使用属性

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'，您应该可以在以后检索它。