假设有一个四维数组idx1,存储另一个五维数组zeros1的5个维度索引。
N,T,H,W = idx1.shape
zeros1 = np.zeros( (N,T,H,W, 256) )
# it is guaranteed that idx1's value <256
我想实现
for n in range(N):
for t in range(T):
for h in range(H):
for w in range(W):
x = idx1[ n,t,h,w ]
zeros1[n,t,h,w,x] = 1
我该如何使用numpy高级索引进行操作。