提交 26dc9fd4 authored 作者: Frederic's avatar Frederic

made available the rebuild_collect_shared function and make the docsting a little better.

上级 e9264ec6
...@@ -20,7 +20,7 @@ from debugmode import DebugMode ...@@ -20,7 +20,7 @@ from debugmode import DebugMode
from profilemode import ProfileMode from profilemode import ProfileMode
from theano.compile.sharedvalue import shared, shared_constructor, SharedVariable from theano.compile.sharedvalue import shared, shared_constructor, SharedVariable
from theano.compile.pfunc import pfunc, Param from theano.compile.pfunc import pfunc, Param, rebuild_collect_shared
from function import function from function import function
...@@ -42,7 +42,7 @@ def rebuild_collect_shared(outputs, ...@@ -42,7 +42,7 @@ def rebuild_collect_shared(outputs,
inputs of the computational graph (or None) inputs of the computational graph (or None)
:type replace: dict :type replace: dict
:param replace: dictionary describing which subgraphs should be :param replace: dictionary describing which subgraphs should be
replaced by what replaced by what. orig_value => new_value
:type updates: dict :type updates: dict
:param updates: dictionary describing updates expressions for shared :param updates: dictionary describing updates expressions for shared
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论