代码之家  ›  专栏  ›  技术社区  ›  Xu Zhang

在numpy中,如何将向量乘以数组/矩阵元素?

  •  4
  • Xu Zhang  · 技术社区  · 8 年前

    我有一个多维数组 a 其形状为(32,3,5,5)和一个数组 v

    1 回复  |  直到 5 年前
        1
  •  2
  •   smci    5 年前

    具有 a v

    a*v[:,None,None,None]
    a*v.reshape(-1, *[1]*3)
    (a.T * v).T
    np.einsum('i...,i->i...', a, v)