Searched refs:apply_along_axis (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/numpy/ |
| D | __init__.py | 34 rot90, select, array_split, choose, size, array_str, apply_along_axis,
|
| D | array_creations.py | 39 apply_along_axis, where, moveaxis
2346 arr = apply_along_axis(func, i, arr, pad_width[i], i, kwargs)
|
| D | array_ops.py | 2260 def apply_along_axis(func1d, axis, arr, *args, **kwargs): function
|
| /third_party/mindspore/tests/st/numpy_native/ |
| D | test_array_ops.py | 1524 mnp_res = mnp.apply_along_axis(mnp.diag, i, mnp_arr)
1525 onp_res = onp.apply_along_axis(onp.diag, i, onp_arr)
1527 mnp_res = mnp.apply_along_axis(lambda x: x[0], 2, mnp_arr)
1528 onp_res = onp.apply_along_axis(lambda x: x[0], 2, onp_arr)
1530 … mnp_res = mnp.apply_along_axis(lambda x, y, offset=0: (x[4] - y)*offset, 2, mnp_arr, 1, offset=3)
1531 … onp_res = onp.apply_along_axis(lambda x, y, offset=0: (x[4] - y)*offset, 2, onp_arr, 1, offset=3)
|