二维NumPy数组中的行外积

Question

我一直在琢磨如何用矢量化来解决这个问题。我有m x n矩阵，其中A是：

A = np.array([A11, A12, ..., A1n], [A21, A22, ..., A2n], ..., [Am1, Am2, ..., Amn]])

我想要一个外积矩阵B，其中大小为(m x n) x n，其中每三行是A每一行的外积，即

计算每行的外积，即A[0, :].T * A[0, :]（我考虑的是a的行向量）。如何使用矢量化实现这一点