cupy.unravel_index#
- cupy.unravel_index(indices, dims, order='C')[source]#
将扁平索引数组转换为坐标数组元组。
- 参数:
indices (cupy.ndarray) – 一个整数数组，其元素是维度为
dims的数组的扁平化版本的索引。dims (整数元组) – 用于展平索引的数组的形状。
order ('C' 或 'F') – 确定索引应按行主序 (C 风格) 还是列主序 (Fortran 风格) 进行索引。
- 返回:
元组中的每个数组与 indices 数组具有相同的形状。
- 返回类型:
ndarray 元组
示例
>>> cupy.unravel_index(cupy.array([22, 41, 37]), (7, 6)) (array([3, 6, 6]), array([4, 5, 1])) >>> cupy.unravel_index(cupy.array([31, 41, 13]), (7, 6), order='F') (array([3, 6, 6]), array([4, 5, 1]))
警告
此函数可能会同步设备。