向量与3D矩阵相乘

| 这让我有些困扰：假设您有一个包含三层的矩阵。是否有一种简单的方法将此矩阵与三个元素的向量相乘，以使第一层（所有元素）与向量的第一个元素相乘，依此类推... 现在我必须使用一个函数来做到这一点：

function out=fun(matrix,vector)

out=matrix;
for k=1:3
    out(:,:,k)=out(:,:,k)*vector(k);
end

有没有一种有效的方法可以在一行中完成此操作而无需功能？

已邀请:

4 个回复

补蹲农界维

一个非常简洁的解决方案是将vector整形为1×1×3矩阵，并使用函数BSXFUN执行逐元素乘法（它将根据需要复制尺寸以匹配两个输入参数的尺寸）：

newMatrix = bsxfun(@times,matrix,reshape(vector,[1 1 3]));

辅奈

有一个名为repmat的matlab函数可以为您提供帮助。

M = [1 2 3]
M * repmat([1 2 3], 3,1)
ans =

 6    12    18
 6    12    18
 6    12    18

M = [1 2 3]
M .* repmat([1 2 3], 3,1)
ans =

 1     4     9
 1     4     9
 1     4     9

取决于您要如何精确地组织矩阵。

容淑阔九

另一种方法是重复向量以按大小匹配矩阵：

out = out.*shiftdim(repmat(vector(:),[1 size(out(:,:,1))]),1)

漂汀拦

除了gnovice的答案外，您还可以沿其他维度复制向量，并直接进行元素明智的乘法。

A=randn(1000,1000,3);%# this is your matrix
vector=[1,2,3];%# this is your vector

[dim1 dim2 ~]=size(A);
replicatedVector=repmat(reshape(vector,1,1,3),[dim1,dim2,1]);
out=A.*replicatedVector;

要回复问题请先登录或注册

向量与3D矩阵相乘

4 个回复

发起人

matlab

vector

matrix

问题状态

向量与3D矩阵相乘

与内容相关的链接

4 个回复

发起人

matlab

vector

matrix

问题状态