下面例子中,加入x為10*3吓懈,x[ix]可以依次訪問x[0][0],x[0][1],x[0][2],x[1][0]...這里不同于多個(gè)for循環(huán)嵌套,for循環(huán)需要一層一個(gè)for循環(huán)棺妓,這里直接適配价涝。
代碼來自cs224n課程作業(yè)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