从3d张量乘以2d矩阵

Question

我有一个三维张量A，size(A)=2 2N。我想得到2x2矩阵的乘积：

result=A(:,:,N)*A(:,:,N-1)*...*A(:,:,1)

这可以通过for循环来完成：

result=A(:,:,N);
for i=(N-1):-1:1
  result=result*A(:,:,i);
end;

但是如何向量化这个操作呢？

Answer 1

这不能用标准的Matlab进行矢量化。然而，一些Mathworks工程师发布了一个名为mtimesx的修改后的mtimes的非常快速的MEX实现，它确实支持您想要做的事情(甚至更多)。请参阅文档中的MTIMESX - Fast Matrix Multiply with Multi-Dimensional Support：

如果A为(2,3,4,5)，B为(3,6,4,5)，则mtimesx(A,B)将生成C(2,6,4,5)，其中C(:,:,i,j) = A(:,:,i,j) * B(:,:,i,j), i=1:4, j=1:5将等同于MATLAB m代码：

C=0(2，6，4，5)；对于n=1:5的m=1:4，C(:，:，m，n) = A(:，:，m，n) * B(:，:，m，n)；end end