python轮询任意维numpy数组每个元素

下面例子中,加入x为10*3,x[ix]可以依次访问x[0][0],x[0][1],x[0][2],x[1][0]...这里不同于多个for循环嵌套,for循环需要一层一个for循环,这里直接适配。
代码来自cs224n课程作业1文件q2_gradcheck.py

    # Iterate over all indexes ix in x to check the gradient.
    it = np.nditer(x, flags=['multi_index'], op_flags=['readwrite'])
    while not it.finished:
        ix = it.multi_index
        # 访问x[ix]
        ......
        it.iternext() # Step to next dimension

你可能感兴趣的:(python轮询任意维numpy数组每个元素)