Home
last modified time | relevance | path

Searched refs:indices_or_sections (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/numpy/
Darray_ops.py1319 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/
Dtest_numpy_ops.py72 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/
Dtest_array_ops.py996 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 …]