Searched refs:indices_or_sections (Results 1 – 3 of 3) sorted by relevance
/third_party/mindspore/mindspore/numpy/ |
D | array_ops.py | 1319 def array_split(x, indices_or_sections, axis=0): argument 1370 return _split(x, indices_or_sections, opname="array_split", axis=axis) 1373 def split(x, indices_or_sections, axis=0): argument 1413 return _split(x, indices_or_sections, opname="split", axis=axis) 1416 def _split(x, indices_or_sections, opname, axis=0): argument 1424 if isinstance(indices_or_sections, int): 1425 if indices_or_sections > length_along_dim: 1427 if opname == "split" or length_along_dim % indices_or_sections == 0: 1428 res = P.Split(axis, indices_or_sections)(x) 1430 num_long_tensor = length_along_dim % indices_or_sections [all …]
|
/third_party/mindspore/tests/ut/python/numpy_native/ |
D | test_numpy_ops.py | 72 a = mnp.split(tensor, indices_or_sections=1) 73 b = mnp.split(tensor, indices_or_sections=3) 74 c = mnp.array_split(tensor, indices_or_sections=1) 75 d = mnp.array_split(tensor, indices_or_sections=3, axis=-1)
|
/third_party/mindspore/tests/st/numpy_native/ |
D | test_array_ops.py | 996 a = mnp.split(input_tensor, indices_or_sections=1) 997 b = mnp.split(input_tensor, indices_or_sections=3) 1002 a = onp.split(input_array, indices_or_sections=1) 1003 b = onp.split(input_array, indices_or_sections=3) 1027 a = mnp.array_split(input_tensor, indices_or_sections=4, axis=2) 1028 b = mnp.array_split(input_tensor, indices_or_sections=3, axis=1) 1029 c = mnp.array_split(input_tensor, indices_or_sections=6) 1034 a = onp.array_split(input_array, indices_or_sections=4, axis=2) 1035 b = onp.array_split(input_array, indices_or_sections=3, axis=1) 1036 c = onp.array_split(input_array, indices_or_sections=6) [all …]
|