Khatri-Rao积(KR积)

输入:A,B两个矩阵,要求两矩阵列数一致

输出:维度为(A行数*B行数)*列数的矩阵。

计算过程:A,B矩阵相同列做kron积运算,并逐列摆放组成结果矩阵。

举例:

A=[1 2;3 4],B=[5 6;7 8]

其中AB均为2*2的矩阵,得到(2*2)*2=4*2的矩阵。

第一列:用A的第一列和B的第一列做kron积运算

[1*5=5

1*7=7

3*5=15

3*7=21]

第二列:用A的第二列和B的第二列做kron积运算

[2*6=12

2*8=16

4*6=24

4*8=32]

最终结果:

[5 12

7 16

15 24

21 32]

Matlab代码:

for k=1:K    %K为输入矩阵的列数

    C(:,k)=kron(A(:,k),B(:,k));
end

猜你喜欢

转载自blog.csdn.net/weixin_38452468/article/details/77245828