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

small docstring clarification following code review.

上级 5de5b4f4
...@@ -65,7 +65,7 @@ class DiffOp(theano.Op): ...@@ -65,7 +65,7 @@ class DiffOp(theano.Op):
def diff(x, n=1, axis=-1): def diff(x, n=1, axis=-1):
"""Calculate the n-th order discrete difference along given axis. """Calculate the n-th order discrete difference along given axis.
The first order difference is given by out[n] = a[n+1] - a[n] The first order difference is given by out[i] = a[i + 1] - a[i]
along the given axis, higher order differences are calculated by along the given axis, higher order differences are calculated by
using diff recursively. Wraping of numpy.diff. using diff recursively. Wraping of numpy.diff.
...@@ -196,6 +196,8 @@ def squeeze(x): ...@@ -196,6 +196,8 @@ def squeeze(x):
:param x: Input data, tensor variable. :param x: Input data, tensor variable.
:return: `x` without its broadcastable dimensions. :return: `x` without its broadcastable dimensions.
.. versionadded:: 0.6
""" """
view = x.dimshuffle([i for i in range(x.ndim) view = x.dimshuffle([i for i in range(x.ndim)
if not x.broadcastable[i]]) if not x.broadcastable[i]])
...@@ -318,6 +320,8 @@ def repeat(x, repeats, axis=None): ...@@ -318,6 +320,8 @@ def repeat(x, repeats, axis=None):
:param repeats: int, scalar or tensor variable. :param repeats: int, scalar or tensor variable.
:param axis: int, optional. :param axis: int, optional.
.. versionadded:: 0.6
""" """
return RepeatOp(axis=axis)(x, repeats) return RepeatOp(axis=axis)(x, repeats)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论