Python-矩阵外部产品
发布于 2021-01-29 17:19:37
给定两个矩阵
A: m * r
B: n * r
我想以产生另一矩阵C: m * n
,每个条目C_ij
是由外积计算的矩阵A_i
和B_j
。
例如,
A: [[1, 2],
[3, 4]]
B: [[3, 1],
[1, 2]]
给
C: [[[3, 1], [[1 ,2],
[6, 2]], [2 ,4]],
[9, 3], [[3, 6],
[12,4]], [4, 8]]]
我可以使用for循环来完成它,例如
for i in range (A.shape(0)):
for j in range (B.shape(0)):
C_ij = np.outer(A_i, B_j)
我想知道是否有矢量化的方法可以加快计算速度?
关注者
0
被浏览
48
1 个回答