convolutions의 for 루프를 arrayfun으로 대체하는 방법은 무엇입니까?

Question

저는 MATLAB에서 3d 행렬의 for-loop를 가지고 있으며, 이것을 arrayfun으로 바꾸고 싶습니다. 그러면 GPU에서 병렬로 실행될 것입니다.

또한 여기에 전체 루프를 제거하는 방법이 느낌,하지만 확실하지 여기에 :)

내 코드, 어떤 제안이 얼마나? 감사.

for i=1:size(obj.W, 3) 
    d(:, :, :, i) = convn(prev_delta(:, :, :, i), ... 
     flipped_W(:, :, i), 'full'); 
end 

Answer 1

입력 배열을 구조체 배열로 변환하고 arrayfun을 적용하는 것이 도움이 되나요?

코드를 테스트하지 않았으므로 시도 할 수 있습니다.

c = cell(size(obj.W, 3), 2); 
p = prev_delta; 
cell_p = squeeze(mat2cell(size(p,1),size(p,2),size(p,3),ones(1,size(p,4)))); 
w = flipped_W; 
cell_w = squeeze(mat2cell(size(w,1), size(w,2),ones(1,size(w,3)))); 

c(:,1) = cell_p; 
c(:,2) = cell_w; 

names = {'p','w'}; 

s = cell2struct(c, names, 2); 

d = arrayfun(@(x)(covn(x.p, x.w, 'full'), s); 

d의 반환 값을 처리해야합니다.