标签: python numpy stack repeat
我有一个矩阵W,我想重复并在第三轴上堆叠x次。这样做最简单的方法是什么?我可以在两个代码行中做到这一点,想知道是否可以在一个代码行中做到这一点:
W
x
np.repeat(W,x,axis=1) W=W.reshape(W.shap[0],W.shap[1],x)
答案 0 :(得分:1)
您可以添加一个新的空轴(使用np。newaxis),然后沿该轴重复。
W = np.repeat(W[:, :, np.newaxis], x, axis=2)