mindspore.numpy.unravel_index
- mindspore.numpy.unravel_index(indices, shape, order='C')[源代码]
将一维索引或一维索引数组转换为坐标数组的tuple。
说明
超出边界的索引会被裁剪到
shape
的边界,而不是引发错误。- 参数:
indices (Union[int, float, bool, list, tuple, Tensor]) - 一个整数数组,其元素是展平后的数组中对应
shape
维度位置的索引。shape (tuple(int)) - 用于将索引转换为坐标的数组shape。
order (Union['C', 'F'], 可选) - 确定索引是否按行主序(C-style)或列主序(Fortran-style)进行处理。默认值:
'C'
。
- 返回:
Tensor,包含坐标数组的tuple,每个数组的shape与索引数组相同。
- 异常:
ValueError - 如果
order
不是‘C’或‘F’。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore.numpy as np >>> print(np.unravel_index([22, 41, 37], (7,6))) (Tensor(shape=[3], dtype=Int32, value= [3, 6, 6]), Tensor(shape=[3], dtype=Int32, value= [4, 5, 1])) >>> print(np.unravel_index([31, 41, 13], (7,6), order='F')) (Tensor(shape=[3], dtype=Int32, value= [3, 6, 6]), Tensor(shape=[3], dtype=Int32, value= [4, 5, 1]))