forked from mindspore-Ecosystem/mindspore
!17098 fix np.array_str, np.size
From: @wangrao124 Reviewed-by: @guoqi1024,@liangchenghui Signed-off-by: @liangchenghui
This commit is contained in:
commit
6bb4bef6c4
|
@ -2207,7 +2207,7 @@ def size(a, axis=None):
|
|||
``Ascend`` ``GPU`` ``CPU``
|
||||
|
||||
Raises:
|
||||
TypeError: If input is not array_like or `axis` is not int or tuple of ints.
|
||||
TypeError: If input is not array_like or `axis` is not int.
|
||||
ValueError: If any axis is out of range or duplicate axes exist.
|
||||
|
||||
Examples:
|
||||
|
@ -2239,7 +2239,7 @@ def array_str(a):
|
|||
Numpy argument `max_line_width`, `precision` and `suppress_small` are not supported.
|
||||
|
||||
Args:
|
||||
a (Union[int, float, bool, list, tuple, Tensor]): Input data.
|
||||
a (Tensor): Input data.
|
||||
|
||||
Returns:
|
||||
String.
|
||||
|
@ -2248,7 +2248,7 @@ def array_str(a):
|
|||
``Ascend`` ``GPU`` ``CPU``
|
||||
|
||||
Raises:
|
||||
TypeError: If input is not array_like.
|
||||
TypeError: If input is not tensor.
|
||||
|
||||
Examples:
|
||||
>>> import mindspore.numpy as np
|
||||
|
@ -2256,7 +2256,8 @@ def array_str(a):
|
|||
>>> np.array_str(x)
|
||||
'[0 1 2 3 4]'
|
||||
"""
|
||||
a = _to_tensor(a)
|
||||
if not isinstance(a, Tensor):
|
||||
_raise_type_error("Expect input to be tensor.")
|
||||
return a.__str__()
|
||||
|
||||
|
||||
|
|
|
@ -891,9 +891,9 @@ def sometrue(a, axis=None, keepdims=False):
|
|||
Examples:
|
||||
>>> import mindspore.numpy as np
|
||||
>>> x = np.array([1, -2.3, 2.1]).astype('float32')
|
||||
>>> output = np.signbit(x)
|
||||
>>> output = np.sometrue(x)
|
||||
>>> print(output)
|
||||
[False True False]
|
||||
True
|
||||
"""
|
||||
if not isinstance(keepdims, int):
|
||||
_raise_type_error("integer argument expected, but got ", keepdims)
|
||||
|
|
Loading…
Reference in New Issue