tensorflow / tensorflow/java

Variable class (like tf.Variable) that supports eager mode

未关闭
#170 1 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看

还没有人认领这个 Issue。

主要语言
Java
星标
928
派生
227
PR 合并指标
30 天内没有已合并 PR

描述

I've been looking at adding a tf.Variable like class, that would work in eager and graph mode (the variable op doesn't work in eager), and register itself in the execution environment. It's not terribly useful until we get eager gradient support, but it's not dependent on it either so there's no reason to wait.

Eager mode is easy, just have a field. For graph, we can store a Variable op, assign to it, but return/expose the return of the assign (it's the new value). The assigns can take the last assign as a control dep.

I've got a draft here I can PR, but I'm not sure how much the type system refactor would affect it. At a minimum I don't see a reason to merge before #160.

贡献指南

打开贡献指南

从这里开始

  1. 先读完整个 Issue,再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库,在一个分支上完成修改。
  4. 提交 Pull Request,并在描述里引用这个 Issue 编号。

调研方向

首先查看链接的 rn_variable 草案,以及 issue #160 中讨论的类型系统重构。检查提议的变量将如何表示 eager 和 graph-mode 状态,以及如何在执行环境中注册自身。当一个类似 tf.Variable 的类能够支持这两种模式,且不依赖 eager 梯度支持时,即表示完成。

由索引模型根据 Issue 内容生成。

评估

技术栈
java
领域
machine-learning
Issue 类型
功能
难度
5/5
预计耗时
一周以上
活跃度
停滞
描述清晰度
基本清楚
新手友好度
25/100

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。