python - 使用 Numpy 在 Python 中平均一组张量

Question

我正在用 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 中使用循环或函数来执行此操作，但我在寻找此问题的答案时遇到了问题。

标签： pythonarraysnumpy

Answer 1

只需这样做：

avg = final_b_vectors.mean(axis = 0)