提交 370b2ff1 authored 作者: Frederic's avatar Frederic

Added missing function in the op contract.

上级 d07083fb
...@@ -145,8 +145,17 @@ following methods: ...@@ -145,8 +145,17 @@ following methods:
Optional. Optional.
This function is needed for shape optimization. If This function is needed for shape optimization. ``shapes`` is a
TODO list with one tuple for each input the Apply node linked to this op
have. Each tuple contain 1 element for each dimensions of the
corresponding inputs. The value is the the corresponding
dimensions shape of the corresponding inputs.
This sound complicated, but this is just the corresponding inputs
shape in symbolic variable.
The function should return a list with one tuple for each output.
Each tuple should contain the corresponding output's shape.
.. function:: make_thunk(node, storage_map, compute_map, no_recycling) .. function:: make_thunk(node, storage_map, compute_map, no_recycling)
...@@ -154,7 +163,11 @@ following methods: ...@@ -154,7 +163,11 @@ following methods:
.. function:: R_op(inputs, eval_points) .. function:: R_op(inputs, eval_points)
TODO Optional.
This function is needed for theano.tensor.Rop to work with this op.
TODO: add more detail.
.. attribute:: default_output .. attribute:: default_output
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论