如何在 Tensorflow C++ 中定义变量的自定义有状态 Op 保存值
How to define a custom stateful Op holding values of variables in Tensorflow C++?
我试图用state定义一个新的Op,然后用python包装它。问题是我希望这个操作在 Session.run(( 之间保存变量(就像在这个操作中初始化一个 int foo,然后在每次运行时将 foo 添加 1。预期的是 foo 比上次运行多 1 个。
我试过REGISTER_OP((。SetIsStateful(( 但它不起作用。有谁知道如何实现这种操作?
谢谢
最后,我找到了如何做到这一点。
根据张量流变量的实现。有状态值可以存储在此处定义的资源管理器中。
简而言之,定义一个包含要存储的状态的结构。并在类资源库中实现虚拟方法。然后通过查找函数还原此有状态结构,该函数从资源管理器管理的容器中检索值。
关键是您需要像这样从上下文中检索资源管理器,而不是定义一个新的资源管理器:
ResourceMgr *rm = context->resource_manager();
相关文章:
- C++映射:具有自定义类的运算符[]不起作用(总是返回0)
- 如何将点击的信号和插槽添加到qt中的自定义按钮中
- C++自定义比较函数
- 如何比较自定义类的std::变体
- std::设置自定义比较器
- 如何正确实现和访问运算符的各种自定义枚举器
- flutter:即使shouldRepaint()返回true,自定义画家也不会重新绘制
- 自定义先决条件对移动分配运算符有效吗
- 使用VS Code和CMake Tools运行自定义命令
- 如何创建从Maya(或类似程序)到虚幻引擎的自定义数据导出插件
- std::ranges::elements_view,用于自定义类似元组的数据
- 用于unique_ptr的有状态自定义删除程序
- C++ grpc::experimental:interceptor 如何从自定义拦截器返回状态和消息
- 如何在 Tensorflow C++ 中定义变量的自定义有状态 Op 保存值
- 如何将自定义分配器的完全相同状态传递给多个容器?
- 从状态内的自定义函数(不是操作)提升 MSM 调用process_event?
- 如何使用预先定义的状态创建自定义快速QML项目
- 我可以使用boost.statecharts自定义对不同正交区域中多个状态的事件做出反应吗?
- 编写自定义纯虚拟处理程序:调用堆栈和寄存器时的状态是什么
- 正确确定自定义绘制的复选框状态