提交 5bed824d authored 作者: jhelie's avatar jhelie

update signatures of choose()

上级 8b596435
...@@ -762,13 +762,13 @@ class _tensor_py_operators(object): ...@@ -762,13 +762,13 @@ class _tensor_py_operators(object):
"""Fill inputted tensor with the assigned value.""" """Fill inputted tensor with the assigned value."""
return theano.tensor.basic.fill(self, value) return theano.tensor.basic.fill(self, value)
def choose(self, a, choices, out=None, mode='raise'): def choose(self, choices, out=None, mode='raise'):
""" """
Construct an array from an index array and a set of arrays to choose Construct an array from an index array and a set of arrays to choose
from. from.
""" """
return theano.tensor.basic.choose(a, choices, out=None, mode='raise') return theano.tensor.basic.choose(self, choices, out=None, mode='raise')
def squeeze(self): def squeeze(self):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论