提交 9bccb0f2 authored 作者: Seon-Wook Park's avatar Seon-Wook Park

Remove out parameter in compress

上级 10f1311e
...@@ -511,7 +511,7 @@ def squeeze(x): ...@@ -511,7 +511,7 @@ def squeeze(x):
return view return view
def compress(condition, x, axis=None, out=None): def compress(condition, x, axis=None):
"""Return selected slices of an array along given axis. """Return selected slices of an array along given axis.
It returns the input tensor, but with selected slices along a given axis It returns the input tensor, but with selected slices along a given axis
...@@ -527,9 +527,6 @@ def compress(condition, x, axis=None, out=None): ...@@ -527,9 +527,6 @@ def compress(condition, x, axis=None, out=None):
.. versionadded:: 0.7 .. versionadded:: 0.7
""" """
# This is done to keep the same function signature then NumPy.
assert out is None
indices = theano.tensor.basic.flatnonzero(condition) indices = theano.tensor.basic.flatnonzero(condition)
return x.take(indices, axis=axis) return x.take(indices, axis=axis)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论