Matlab Vectorize

Question

У меня есть матрица вероятности (glcm) размером 256x256x20. Я изменил матрицу на 65536x20, так что я могу исключить один цикл (по третьему измерению).

Я хочу сделать следующий расчет.

for y = 1:256 
    for x = 1:256 
     if (ismember((x + y),(2:2*256))) 
      p_xplusy((x+y),:) = p_xplusy((x+y),:) + glcm(((y-1)*256+x),:); 
     end 
    end 
end 

Так p_xplusy будет 511x20 матрица, каждый элемент является суммой диагонали матрицы NxN к югу (где п принадлежит 1: 256) исходной матрицы 256x256x20.

Этот код блокирует мою программу неэффективно, и я хочу векторизовать этот цикл. Любая помощь будет оценена по достоинству.

Answer 1

Поскольку ваш if оператор просто проверить ли x+y меньше или равно 256, просто заставить его быть всегда, и удалить лишние петли:

for y = 1:256 
    for x = 1:256-y 
     p_xplusy((x+y),:) = p_xplusy((x+y),:) + glcm(((y-1)*256+x),:); 
    end 
end 

Это должно обеспечить заметное ускорение для вашего код.

Answer 2

Вы можете уменьшить сложность от O(n^2) к O(2*n) и тем самым повысить эффективность выполнения -

N = 256; 
for k1 = 1:N 
    idx_glcm = k1:N-1:N*(k1-1)+1; 
    p_xplusy(k1+1,:) = p_xplusy(k1+1,:) + sum(glcm(idx_glcm,:),1); 
end 

for k1 = 2:N 
    idx_glcm = k1*N:N-1:N*(N-1)+k1; 
    p_xplusy(N+k1,:) = p_xplusy(N+k1,:) + sum(glcm(idx_glcm,:),1); 
end 

Некоторые быстрые тесты во время выполнения, кажется, подтверждают нашу теорию эффективности тоже.