提交 edfb8105 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Document the optional arguments facilites for COp and rename to a

clearer name the properties.
上级 e83f5a03
...@@ -893,6 +893,12 @@ If you pass a function name to the ``__init__()`` method of the ...@@ -893,6 +893,12 @@ If you pass a function name to the ``__init__()`` method of the
types for the argument is dependant on the Types (that is types for the argument is dependant on the Types (that is
theano Types) of your inputs and outputs. theano Types) of your inputs and outputs.
* You can sepcify the number of inputs and outputs for your op
by setting the `_cop_num_inputs` and `_cop_num_outputs`
attributes on your op. The main function will always be
called with that number of arguments, using NULL to fill in
for missing values at the end.
For example, the main C function of an op that takes two TensorTypes For example, the main C function of an op that takes two TensorTypes
(which has ``PyArrayObject *`` as its C type) as inputs and returns (which has ``PyArrayObject *`` as its C type) as inputs and returns
both their sum and the difference between them would have four both their sum and the difference between them would have four
......
...@@ -1435,11 +1435,11 @@ class COp(Op): ...@@ -1435,11 +1435,11 @@ class COp(Op):
# function. The argstring will be of format : # function. The argstring will be of format :
# "input0, input1, input2, &output0, &output1" # "input0, input1, input2, &output0, &output1"
inp = list(inp) inp = list(inp)
numi = getattr(self, '_cop_numi', len(inp)) numi = getattr(self, '_cop_num_inputs', len(inp))
while len(inp) < numi: while len(inp) < numi:
inp.append('NULL') inp.append('NULL')
out = ["&%s" % o for o in out] out = ["&%s" % o for o in out]
numo = getattr(self, '_cop_numo', len(out)) numo = getattr(self, '_cop_num_outputs', len(out))
while len(out) < numo: while len(out) < numo:
out.append('NULL') out.append('NULL')
return ", ".join(inp + out) return ", ".join(inp + out)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论