用Numpy平均Python中的张量数组

我正在做一个关于StyleGans的项目，实际上我对Python或Numpy不是很了解

我有一个向量数组

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中使用循环或函数有更好的方法，但我在搜索此问题的答案时遇到问题。

只需执行以下操作：

avg = final_b_vectors.mean(axis = 0)

---

我建议您阅读有关轴、维度、索引和多维数组基本函数的numpy文档。如果你熟悉它们，那将非常有帮助。