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

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))这将给出所需的结果。