diff --git a/pytato/array.py b/pytato/array.py index c546d2f5f523d7f8579cd4033475fc42a3398039..25ffc525e56136e155b51d7a46c7ac423e846ef3 100644 --- a/pytato/array.py +++ b/pytato/array.py @@ -354,6 +354,8 @@ class Array(Taggable): .. method:: __rxor__ .. method:: __abs__ .. method:: conj + .. automethod:: all + .. automethod:: any .. autoattribute:: real .. autoattribute:: imag @@ -611,6 +613,20 @@ class Array(Taggable): # expected "Union[int, Sequence[int]]" return pt.reshape(self, shape, order=order) # type: ignore + def all(self, axis: int = 0) -> ArrayOrScalar: + """ + Equivalent to :func:`pytato.all`. + """ + import pytato as pt + return pt.all(self, axis) + + def any(self, axis: int = 0) -> ArrayOrScalar: + """ + Equivalent to :func:`pytato.any`. + """ + import pytato as pt + return pt.any(self, axis) + # }}}