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,:,:); 

這並不意味着它會更快，但是，當你計算多次多個值。