一维索引的张量扩展到多维索引

Question

我想从任何维度的一维索引数组中获得多维数组的功能。 例如，我有

x=np.array([1,2,3])
def extend_array(x,d)

如果我将d = 2放进去，那么得到的数组就是

[[1,1],
 [1,2],
 [1,3],
 [2,1],
 [2,2],
 [2,3],
 [3,1],
 [3,2],
 [3,3]]

或输入可以是（n，d）而不是（x，d），以便n表示

np.array(range(1,n+1))

我认为有一个功能，但是我找不到。 您能帮我解决这个问题吗？

Answer 1

from itertools import product
t=list(product(x,repeat=d))

这将提供所需的结果。