下面例子中,加入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