在Python中矢量化多个外部产品?

时间:2016-08-04 17:38:49

标签: python matrix vectorization

我看过this question here看起来很相似,但我的问题可能更简单。

我的矩阵 A 的大小为[N x C],矩阵 X 的大小为[N x D]

对于 A 中的每个 n 行,使用 X n 行计算它的外部产品>。每个外部产物将产生尺寸为[C x D]的基质。然后,将所有这些基质总结在一起以获得最终矩阵。

在Python中有一种简单的非循环方式吗?

谢谢!

1 个答案:

答案 0 :(得分:1)

取第n行外:element(c,d)是A [n,c] * X [n,d]。现在对所有n进行求和,得到Sum_n A [n,c] * X [n,d]正是(AT.X)[c,d]