support axis is None for all and any interface in graph mode.

This commit is contained in:
buxue 2020-08-17 17:53:02 +08:00
parent f41ca6b5c6
commit 09f1a4bbaf
2 changed files with 6 additions and 3 deletions

View File

@ -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)

View File

@ -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()