第三维 numpy 上的堆栈矩阵

Stack matrix on the third dimension numpy

我有一个矩阵 W,我想将其重复并在第三个轴上堆叠 x 次。最简单的方法是什么?我可以在两个代码行中完成,想知道是否可以在一个代码行中完成:

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

您可以添加一个新的空轴(使用 np.newaxis),然后沿着该轴重复。

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