提交 266b5baa authored 作者: Frederic's avatar Frederic

better doc.

上级 d9052e5c
......@@ -428,6 +428,8 @@ def images2neibs(ten4, neib_shape, neib_step=None, mode='valid'):
Reshapes the input as a 2D tensor where each row is an
pooling example. Pseudo-code of the output:
.. code-block:: python
idx = 0
for i in xrange(list 1 dim)
for j in xrange(list 2 dim)
......@@ -447,11 +449,11 @@ def neibs2images(neibs, neib_shape, original_shape, mode='valid'):
"""
Inverse of images2neib.
neibs : matrix like the one obtained by images2neib
neib_shape : neib_shape that was used in images2neib
original_shape : original shape of the 4d tensor given to images2neib
:param neibs: matrix like the one obtained by images2neib
:param neib_shape: neib_shape that was used in images2neib
:param original_shape: original shape of the 4d tensor given to images2neib
Return a 4d tensor of shape `original_shape`.
:return: Return a 4d tensor of shape `original_shape`.
"""
neibs = T.as_tensor_variable(neibs)
neib_shape = T.as_tensor_variable(neib_shape)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论