代码之家  ›  专栏  ›  技术社区  ›  havakok

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

  •  1
  • havakok  · 技术社区  · 6 年前

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

    np.repeat(W,x,axis=1)
    W=W.reshape(W.shap[0],W.shap[1],x)
    
    1 回复  |  直到 6 年前
        1
  •  1
  •   Alex bGoode    6 年前

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

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