forked from mindspore-Ecosystem/mindspore
support axis is None for all and any interface in graph mode.
This commit is contained in:
parent
f41ca6b5c6
commit
09f1a4bbaf
|
@ -44,6 +44,8 @@ def all_(x, axis=(), keep_dims=False):
|
|||
Tensor, has the same data type as x.
|
||||
"""
|
||||
|
||||
if axis is None:
|
||||
axis = ()
|
||||
reduce_all = P.ReduceAll(keep_dims)
|
||||
return reduce_all(x, axis)
|
||||
|
||||
|
@ -60,7 +62,8 @@ def any_(x, axis=(), keep_dims=False):
|
|||
Returns:
|
||||
Tensor, has the same data type as x.
|
||||
"""
|
||||
|
||||
if axis is None:
|
||||
axis = ()
|
||||
reduce_any = P.ReduceAny(keep_dims)
|
||||
return reduce_any(x, axis)
|
||||
|
||||
|
|
|
@ -28,8 +28,8 @@ def test_all_and_any_of_tensor_in_graph():
|
|||
def construct(self, x):
|
||||
all_ = x.all()
|
||||
any_ = x.any()
|
||||
all_0 = x.all(0, True)
|
||||
any_0 = x.any(0, True)
|
||||
all_0 = x.all(None, True)
|
||||
any_0 = x.any(None, True)
|
||||
return all_, any_, all_0, any_0
|
||||
|
||||
net = Net()
|
||||
|
|
Loading…
Reference in New Issue