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

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))

这将给出所需的结果。