Home
last modified time | relevance | path

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

/third_party/mindspore/tests/st/numpy_native/
Dtest_array_ops.py219 a = mnp.rollaxis(input_tensor, 0, 1)
220 b = mnp.rollaxis(input_tensor, 0, 2)
221 c = mnp.rollaxis(input_tensor, 2, 1)
222 d = mnp.rollaxis(input_tensor, 2, 2)
223 e = mnp.rollaxis(input_tensor, 0)
224 f = mnp.rollaxis(input_tensor, 1)
229 a = onp.rollaxis(input_array, 0, 1)
230 b = onp.rollaxis(input_array, 0, 2)
231 c = onp.rollaxis(input_array, 2, 1)
232 d = onp.rollaxis(input_array, 2, 2)
[all …]
/third_party/mindspore/mindspore/numpy/
D__init__.py29 from .array_ops import (transpose, expand_dims, squeeze, rollaxis, swapaxes, reshape,
Darray_ops.py140 def rollaxis(x, axis, start=0): function