提交 af1fb499 authored 作者: ChienliMa's avatar ChienliMa

Swap vairable in Ins and improve docs

上级 2e7c2590
...@@ -545,7 +545,11 @@ class Function(object): ...@@ -545,7 +545,11 @@ class Function(object):
""" """
Copy this function. Copied function will have separated maker and Copy this function. Copied function will have separated maker and
fgraph with original function. User can choose whether to separate fgraph with original function. User can choose whether to separate
storage by changing the share_memory arguments storage by changing the share_memory arguments.
Note:
We reuse original Out instances but In instances, therefore variables
in Ins and Outs are not in the same graph. Please avoid using variables
is Ins and Outs to reinitialize another Function or FunctionGraoh.
--------------------- ---------------------
Params: Params:
share_memory -- { boolean } Default is False. When True, two share_memory -- { boolean } Default is False. When True, two
...@@ -554,6 +558,10 @@ class Function(object): ...@@ -554,6 +558,10 @@ class Function(object):
storages( see method __copy__() ) and same maker. If two functions storages( see method __copy__() ) and same maker. If two functions
share memory and allow_gc=False, this will increase executing speed share memory and allow_gc=False, this will increase executing speed
and save memory. and save memory.
swap -- { dict } Dictionary<String, theano.SharedVariable> that
map old SharedVariable's name to new SharedVariable. Default is
None.
--------------------- ---------------------
Returns: Returns:
func -- Copied theano.Function func -- Copied theano.Function
...@@ -607,7 +615,7 @@ class Function(object): ...@@ -607,7 +615,7 @@ class Function(object):
accept_inplace=maker.accept_inplace).create( accept_inplace=maker.accept_inplace).create(
input_storage, storage_map=new_storage_map) input_storage, storage_map=new_storage_map)
# Share immutable and constant input storage # Share immutable and constant input's storage
for in_ori, in_cpy, ori, cpy in zip(maker.inputs, f_cpy.maker.inputs, for in_ori, in_cpy, ori, cpy in zip(maker.inputs, f_cpy.maker.inputs,
self.input_storage, self.input_storage,
f_cpy.input_storage): f_cpy.input_storage):
...@@ -616,6 +624,10 @@ class Function(object): ...@@ -616,6 +624,10 @@ class Function(object):
cpy.data = ori.data cpy.data = ori.data
in_cpy.value = in_ori.value in_cpy.value = in_ori.value
# swap SharedVariable in In instances
if in_cpy.variable.name in swap.keys():
in_cpy.variable = swap[in_cpy.variable.name]
return f_cpy return f_cpy
def _swapSV(self, swap, ins, fg_cpy, memo): def _swapSV(self, swap, ins, fg_cpy, memo):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论