Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/numpy/
Darray_creations.py32 _expand, _to_tensor, _slice_along_axis, _callable
2175 pad_before = stat_op(_slice_along_axis(arr, i, 0, stat_length[i][0]), i)
2177 pad_after = stat_op(_slice_along_axis(arr, i, shape[i]-stat_length[i][1], shape[i]), i)
2202 tensor_with_pad += (_slice_along_axis(arr, i, shape[i]-padsize_before, shape[i]),)
2205 tensor_with_pad += (_slice_along_axis(arr, i, 0, padsize_after),)
2217 left_value = _slice_along_axis(arr, i, 0, 1)
2218 right_value = _slice_along_axis(arr, i, shape[i]-1, shape[i])
2227 pad_after = (_slice_along_axis(pad_after, i, 1, pad_width[i][1]+1),)
2238 edge_before = _slice_along_axis(arr, idx, 0, 1)
2250 curr_pad = _slice_along_axis(arr, idx, endpoint_adder, endpoint)
[all …]
Dutils.py164 def _slice_along_axis(f, axis, slice_start, slice_end): function
Dmath_ops.py3896 def _slice_along_axis(f, axis, slice_start, slice_end): function
3920 upper_edge = _slice_along_axis(f, axis, 1, 2) - _slice_along_axis(f, axis, 0, 1)
3921 lower_edge = _slice_along_axis(f, axis, end-1, end) - _slice_along_axis(f, axis, end-2, end-1)
3925 middle = (_slice_along_axis(f, axis, 2, end) - _slice_along_axis(f, axis, 0, end-2)) * 0.5