MAtlab - 是否有可能用一个convn替换这个for-loop？

Question

我有以下代码，其中数据是[128 64 10]的矩阵，W是[8 64 50]的矩阵。我正在计算m

的卷积

for j = 1 : size(obj.W, 3) 
  output(:, :, :, j) = convn(data, obj.W(:, :, j), 'valid');
end

Answer 1

是：

output = convn(data, reshape(obj.W, [8 64 1 50]), 'full');
output = output(8:end-7,64,:,:);

但这并不意味着它会更快，因为你计算的值会增加很多倍。