如何定义TysFraseC++中自定义变量的状态保持值？

Question

我试图用state定义一个新的Op，然后用python包装它。问题是我希望这个运算在会话.运行（）（比如在这个操作中初始化一个intfoo，然后每次运行时将foo加1。预期的结果是foo比上一次运行多1次。）

我尝试过REGISTER_OP（）.SetIsStateful（），但它不起作用。有人知道如何实现这种操作吗？在

谢谢

Answer 1

最后，我找到了如何做到这一点。在

根据tensorflow的实现变量有状态值可以存储在资源管理器定义的here中。在

简单地说，定义一个包含要存储的状态的结构。并在类资源库中实现了虚拟方法。然后通过查找函数还原这个有状态的结构，该函数从资源管理器管理的容器中检索值。在

重点是，您需要像这样从上下文中检索ResourceManager，而不是定义一个新的：

ResourceMgr *rm = context->resource_manager();