代码之家  ›  专栏  ›  技术社区  ›  Some Guy Trying To Do Math

用numpython求张量数组的平均值

  •  0
  • Some Guy Trying To Do Math  · 技术社区  · 4 年前

    我有一个向量数组

    final_b_vectors = np.array(final_b_vectors) // final_b_vectors is an ndarray with shape (14, 18, 512)
    

    我想找出数组的平均值

    我目前使用

    averageB = final_b_vectors[0] + final_b_vectors[1] + final_b_vectors[2] + final_b_vectors[3] + final_b_vectors[4] + final_b_vectors[5] + final_b_vectors[6] + final_b_vectors[7] + final_b_vectors[8] + final_b_vectors[9] + final_b_vectors[10] + final_b_vectors[11]  + final_b_vectors[12] + final_b_vectors[13]
    averageB = averageB /14
    

    我确信有一种更好的方法可以在numpy中使用循环或函数来实现这一点,但是我在寻找这个问题的答案时遇到了问题。

    1 回复  |  直到 4 年前
        1
  •  1
  •   swag2198    4 年前

    只需这样做:

    avg = final_b_vectors.mean(axis = 0)