Searched refs:take_along_axis (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/numpy/ |
| D | __init__.py | 33 flip, flipud, fliplr, hsplit, dsplit, take_along_axis, take, repeat,
|
| D | array_ops.py | 1745 def take_along_axis(arr, indices, axis): function
|
| /third_party/mindspore/tests/st/numpy_native/ |
| D | test_array_ops.py | 1143 a = mnp.take_along_axis(x, arrs[1], axis=None)
1144 b = mnp.take_along_axis(x, arrs[2], axis=1)
1145 c = mnp.take_along_axis(x, arrs[3], axis=-1)
1146 d = mnp.take_along_axis(x, arrs[4], axis=0)
1152 a = onp.take_along_axis(x, arrs[1], axis=None)
1153 b = onp.take_along_axis(x, arrs[2], axis=1)
1154 c = onp.take_along_axis(x, arrs[3], axis=-1)
1155 d = onp.take_along_axis(x, arrs[4], axis=0)
|