提交 ac974a81 authored 作者: Frederic's avatar Frederic

Fix copy-pasted docstring.

上级 89f7c537
...@@ -18,7 +18,7 @@ def register_view_op_c_code(type, code, version=()): ...@@ -18,7 +18,7 @@ def register_view_op_c_code(type, code, version=()):
:param typ: A Theano type. It must be the Theano class itself and not an :param typ: A Theano type. It must be the Theano class itself and not an
instance of the class. instance of the class.
:param code: C code that deep copies the Theano type 'typ'. :param code: C code that return a view for the Theano type 'typ'.
Use %(iname)s and %(oname)s for the input and output C Use %(iname)s and %(oname)s for the input and output C
variable names respectively. variable names respectively.
:param version: A number indicating the version of the code, for cache. :param version: A number indicating the version of the code, for cache.
...@@ -193,7 +193,8 @@ def register_shape_c_code(type, code, version=()): ...@@ -193,7 +193,8 @@ def register_shape_c_code(type, code, version=()):
:param typ: A Theano type. It must be the Theano class itself and not an :param typ: A Theano type. It must be the Theano class itself and not an
instance of the class. instance of the class.
:param code: C code that deep copies the Theano type 'typ'. :param code: C code that return a vector representing the shape
for the Theano type 'typ'.
Use %(iname)s and %(oname)s for the input and output C Use %(iname)s and %(oname)s for the input and output C
variable names respectively. variable names respectively.
:param version: A number indicating the version of the code, for cache. :param version: A number indicating the version of the code, for cache.
...@@ -372,7 +373,7 @@ def register_shape_i_c_code(typ, code, version=()): ...@@ -372,7 +373,7 @@ def register_shape_i_c_code(typ, code, version=()):
:param typ: A Theano type. It must be the Theano class itself and not an :param typ: A Theano type. It must be the Theano class itself and not an
instance of the class. instance of the class.
:param code: C code that deep copies the Theano type 'typ'. :param code: C code that get the shape of dimensions %(i)s for the Theano type 'typ'.
Use %(iname)s and %(oname)s for the input and output C Use %(iname)s and %(oname)s for the input and output C
variable names respectively. variable names respectively.
:param version: A number indicating the version of the code, for cache. :param version: A number indicating the version of the code, for cache.
...@@ -479,14 +480,16 @@ def as_op(itypes, otypes, infer_shape=None): ...@@ -479,14 +480,16 @@ def as_op(itypes, otypes, infer_shape=None):
def register_rebroadcast_c_code(typ, code, version=()): def register_rebroadcast_c_code(typ, code, version=()):
""" Tell Rebroadcast how to generate C code for a Theano Type """Tell Rebroadcast how to generate C code for a Theano Type
:param typ: A Theano type. It must be the Theano class itself and not an :param typ: A Theano type. It must be the Theano class itself and not an
instance of the class. instance of the class.
:param code: C code that deep copies the Theano type 'typ'.
Use %(iname)s and %(oname)s for the input and output C :param code: C code that check if the dimensions %(axis) is of
variable names respectively. shape 1 for the Theano type 'typ'. Use %(iname)s and
%(axis)s for the axis that we need to check. %(oname)s for the input and output C variable names
respectively. %(axis)s for the axis that we need to
check. This code is put in a loop for all axis
:param version: A number indicating the version of the code, for cache. :param version: A number indicating the version of the code, for cache.
""" """
Rebroadcast.c_code_and_version[typ] = (code, version) Rebroadcast.c_code_and_version[typ] = (code, version)
...@@ -619,10 +622,10 @@ def register_specify_shape_c_code(typ, code, version=(), ...@@ -619,10 +622,10 @@ def register_specify_shape_c_code(typ, code, version=(),
:param typ: A Theano type. It must be the Theano class itself and not an :param typ: A Theano type. It must be the Theano class itself and not an
instance of the class. instance of the class.
:param code: C code that deep copies the Theano type 'typ'. :param code: C code that check the shape and return a view for the Theano type 'typ'.
Use %(iname)s and %(oname)s for the input and output C Use %(iname)s and %(oname)s for the input and output C
variable names respectively. variable names respectively.
%(axis)s for the axis that we need to check. %(axis)s for the axis that we need to check in a loop on iname.ndim.
:param version: A number indicating the version of the code, for cache. :param version: A number indicating the version of the code, for cache.
:param c_support_code_apply: extra code. :param c_support_code_apply: extra code.
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论