第三维上的堆栈矩阵numpy [重复]

Question

这个问题在这里已有答案：

• Numpy: stack duplicates of matrix in new dimension [duplicate] 1回答

我有一个矩阵W我想重复并在第三轴上叠加x次。最简单的方法是什么？我可以在两个代码行中完成它，想知道是否可以在一个代码行中完成：

np.repeat(W,x,axis=1)
W=W.reshape(W.shap[0],W.shap[1],x)

Answer 1

您可以添加一个新的空轴（使用np.newaxis），然后沿此轴重复。

W = np.repeat(W[:, :, np.newaxis], x, axis=2)