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

Remove out parameter in compress

上级 10f1311e
......@@ -511,7 +511,7 @@ def squeeze(x):
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.
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):
.. versionadded:: 0.7
"""
# This is done to keep the same function signature then NumPy.
assert out is None
indices = theano.tensor.basic.flatnonzero(condition)
return x.take(indices, axis=axis)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论