创建填充数组本身索引的numpy数组

Question

我正在尝试创建一个存储数组本身索引的np数组。例如，在一维情况下，我想拥有

[[0],[1],[2],[3]]

在2d 2x2阵列的情况下，我想拥有

[[[0,0], [0,1]], [[1,0], [1,1]]]

到目前为止，我发现的唯一方法是一堆循环-考虑到我必须处理的数组有100万个以上的条目，效率不是很高：

    idx_array = np.empty(shape=kernel_shape+[4])

    for ker_x in range(kernel_shape[0]):
        for ker_y in range(kernel_shape[1]):
            for in_ch in range(kernel_shape[2]):
                for out_ch in range(kernel_shape[3]):
                    idx_array[ker_x][ker_y][in_ch][out_ch] = [ker_x, ker_y, in_ch, out_ch]

是否有更好的方法来实现这一点？

谢谢！