在numpy中使用非平凡的索引向量化五元组循环

Question

我有一系列嵌套for循环，用于填充rank-3数组，如下所示：

S = ...  # Some number
N = ...  # Some other number
an_array = np.zeros((S, N * N, N * N))
for s in range(S):
    for i in range(N):
        for a in range(N):
            for j in range(N):
                for b in range(N):
                    an_array[s, i * N + a, j * N + b] = target[s, i, a, j, b]

这相当于在数组的最右边的两个维度中具有基数N的索引，其中数组的元素被索引为an_array[s, ia, jb]，除了索引在基数10中转换为与numpy一起工作。

如何在不使用for循环的情况下以矢量化方式填充数组？

由于