python - 如何将形状从（10,50）变为（10，）？

Question

所以我的矩阵 A 形状是 (10,)，矩阵 B 形状是 (10,50)。如何将矩阵 B 的形状从 (10,50) 更改为 (10,)，以便可以用矩阵 A 减去它？

标签： pythonnumpymatrix

Answer 1

您可以通过广播 theta到二维数组（此处为 shape (10, 1)）来做到这一点：

>>> theta[:,None] - gamma