提交 6a9a353f authored 作者: Jeremiah Lowin's avatar Jeremiah Lowin

add sort method

上级 90eb96fe
...@@ -1796,6 +1796,11 @@ class _tensor_py_operators: ...@@ -1796,6 +1796,11 @@ class _tensor_py_operators:
"""See `theano.tensor.argmax`""" """See `theano.tensor.argmax`"""
return argmax(self, axis, keepdims=keepdims) return argmax(self, axis, keepdims=keepdims)
def sort(self, axis=-1, kind='quicksort', order=None):
"""See `theano.tensor.sort`"""
from theano.tensor.sort import sort
return sort(self, axis, kind, order)
def argsort(self, axis=-1, kind='quicksort', order=None): def argsort(self, axis=-1, kind='quicksort', order=None):
"""See `theano.tensor.argsort`""" """See `theano.tensor.argsort`"""
from theano.tensor.sort import argsort from theano.tensor.sort import argsort
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论