提交 0f00c10f authored 作者: Iban Harlouchet's avatar Iban Harlouchet

numpydoc for theano/compile/function.py

上级 e6ecae1c
"""Define the `function` function """
Define the `function` function.
""" """
import six.moves.cPickle as pickle import six.moves.cPickle as pickle
import logging import logging
...@@ -23,8 +25,9 @@ def function_dump(filename, inputs, outputs=None, mode=None, updates=None, ...@@ -23,8 +25,9 @@ def function_dump(filename, inputs, outputs=None, mode=None, updates=None,
no_default_updates=False, accept_inplace=False, name=None, no_default_updates=False, accept_inplace=False, name=None,
rebuild_strict=True, allow_input_downcast=None, profile=None, rebuild_strict=True, allow_input_downcast=None, profile=None,
on_unused_input=None): on_unused_input=None):
"""This is helpful to make a reproducable case for problem during """
Theano compilation. This is helpful to make a reproducable case for problem during Theano
compilation.
Ex: Ex:
...@@ -65,73 +68,62 @@ def function(inputs, outputs=None, mode=None, updates=None, givens=None, ...@@ -65,73 +68,62 @@ def function(inputs, outputs=None, mode=None, updates=None, givens=None,
""" """
Return a callable object that will calculate `outputs` from `inputs`. Return a callable object that will calculate `outputs` from `inputs`.
:type inputs: list of either Variable or Param instances. Parameters
:param inputs: function parameters, these are not allowed to be shared ----------
variables inputs : list of either Variable or Param instances.
Function parameters, these are not allowed to be shared variables.
:type outputs: list or dict of Variables or Out instances. If it is a outputs : list or dict of Variables or Out instances.
dict, the keys must be strings If it is a dict, the keys must be strings. Expressions to compute.
:param outputs: expressions to compute mode : string or `Mode` instance.
Compilation mode.
:type mode: string or `Mode` instance. updates : iterable over pairs (shared_variable, new_expression). List, tuple
:param mode: compilation mode or OrderedDict.
Updates the values for SharedVariable inputs according to these
:type updates: iterable over pairs (shared_variable, new_expression). expressions.
List, tuple or OrderedDict. givens : iterable over pairs (Var1, Var2) of Variables. List, tuple or dict.
:param updates: update the values for SharedVariable inputs The Var1 and Var2 in each pair must have the same Type.
according to these expressions Specific substitutions to make in the computation graph (Var2 replaces
Var1).
:type givens: iterable over pairs (Var1, Var2) of Variables. List, no_default_updates: either bool or list of Variables
tuple or dict. The Var1 and Var2 in each pair must If True, do not perform any automatic update on Variables. If False
have the same Type. (default), perform them all. Else, perform automatic updates on all
:param givens: specific substitutions to make in the computation Variables that are neither in "updates" nor in "no_default_updates".
graph (Var2 replaces Var1). name : str
An optional name for this function. The profile mode will print the time
:type no_default_updates: either bool or list of Variables spent in this function.
:param no_default_updates: if True, do not perform any automatic rebuild_strict : bool
update on Variables. If False (default), perform them True (Default) is the safer and better tested setting, in which case
all. Else, perform automatic updates on all Variables that are `givens` must substitute new variables with the same Type as the
neither in "updates" nor in "no_default_updates". variables they replace.
False is a you-better-know-what-you-are-doing setting, that permits
:param name: an optional name for this function. The profile mode `givens` to replace variables with new variables of any Type.
will print the time spent in this function. The consequence of changing a Type is that all results depending on that
variable may have a different Type too (the graph is rebuilt from inputs
:param rebuild_strict: True (Default) is the safer and better to outputs). If one of the new types does not make sense for one of the
tested setting, in which case `givens` must substitute new Ops in the graph, an Exception will be raised.
variables with the same Type as the variables they replace. allow_input_downcast: bool or None
False is a you-better-know-what-you-are-doing setting, that True means that the values passed as inputs when calling the function
permits `givens` to replace variables with new variables of can be silently downcasted to fit the dtype of the corresponding
any Type. The consequence of changing a Type is that all Variable, which may lose precision. False means that it will only be
results depending on that variable may have a different Type cast to a more general, or precise, type. None (default) is almost like
too (the graph is rebuilt from inputs to outputs). If one of False, but allows downcasting of Python float scalars to floatX.
the new types does not make sense for one of the Ops in the profile: None, True, or ProfileStats instance
graph, an Exception will be raised. Accumulate profiling information into a given ProfileStats instance.
If argument is `True` then a new ProfileStats instance will be used.
:type allow_input_downcast: Boolean or None This profiling object will be available via self.profile.
:param allow_input_downcast: True means that the values passed as on_unused_input
inputs when calling the function can be silently downcasted to What to do if a variable in the 'inputs' list is not used in the graph.
fit the dtype of the corresponding Variable, which may lose Possible values are 'raise', 'warn', 'ignore' and None.
precision. False means that it will only be cast to a more
general, or precise, type. None (default) is almost like Returns
False, but allows downcasting of Python float scalars to -------
floatX. Function instance
A callable object that will compute the outputs (given the inputs) and
:type profile: None, True, or ProfileStats instance update the implicit function arguments according to the `updates`.
:param profile: accumulate profiling information into a given
ProfileStats instance. If argument is `True` then a new Notes
ProfileStats instance will be used. This profiling object -----
will be available via self.profile. Regarding givens: Be careful to make sure that these
:param on_unused_input: What to do if a variable in the 'inputs'
list is not used in the graph. Possible values are 'raise',
'warn', 'ignore' and None.
:rtype: Function instance
:returns: a callable object that will compute the outputs (given
the inputs) and update the implicit function arguments
according to the `updates`.
:note: Regarding givens: Be careful to make sure that these
substitutions are independent--behaviour when Var1 of one pair substitutions are independent--behaviour when Var1 of one pair
appears in the graph leading to Var2 in another expression is appears in the graph leading to Var2 in another expression is
undefined. Replacements specified with givens are different undefined. Replacements specified with givens are different
...@@ -214,6 +206,7 @@ def function(inputs, outputs=None, mode=None, updates=None, givens=None, ...@@ -214,6 +206,7 @@ def function(inputs, outputs=None, mode=None, updates=None, givens=None,
was easier to develop the VM in Python then translate it to C instead was easier to develop the VM in Python then translate it to C instead
of just writing it in C from scratch. of just writing it in C from scratch.
CVM stands for C Virtual Machine. CVM stands for C Virtual Machine.
""" """
if isinstance(outputs, dict): if isinstance(outputs, dict):
output_items = list(outputs.items()) output_items = list(outputs.items())
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论