Векторизация многомерного массива населения на октаву - PullRequest
Новая
       122

Векторизация многомерного массива населения на октаву

1 голос
/ 30 мая 2019

Это мой фактический код: 

Lr = linspace(100,300,10);
vi = linspace(10,30,10);
vf = linspace(10,30,10);
b = linspace(0.5,1.2,10);
h = linspace(0.3,0.8,10);
Rc = [1000, 1500, 2000, 2500, 3000, 3500;
      29, 22.4, 17.3, 13.4, 10.4, 8];
rti = randi(5, 10, 1, 10, 10, 10, 10);
for kk = 1:rows(Lr)
  for jj = 1:length(vi)
    for ll = 1:length(vf)
      for mm = 1:length(b)
        for nn = 1:length(h)
          ratt(kk,1,jj,ll,mm,nn) = Rc(2,rti(kk,1,jj,ll,mm,nn));
        endfor
      endfor
    endfor
  endfor
endfor

В этом примере все векторы имеют длину 10, но в реальном коде они могут быть от 1 до 100. Как я могу удалить все эти петли?

1 Ответ

0 голосов
/ 30 мая 2019
  1. Получить все значения в ratt из Rc в виде 1D-массива.
  2. Измените ratt в 6D-массив, как в исходном коде.

Код: 

ratt = Rc(2, rti(1:rows(Lr), 1, 1:length(vi), 1:length(vf), 1:length(b), 1:length(h)));
ratt = reshape(ratt, [rows(Lr), 1, length(vi), length(vf), length(b), length(h)]);

Тестирование: Чтобы проверить равенство и скорость векторизованных и не векторизованных кодов, я написал следующий код: 

Lr = linspace(100,300,10);
vi = linspace(10,30,10);
vf = linspace(10,30,10);
b = linspace(0.5,1.2,10);
h = linspace(0.3,0.8,10);
Rc = [1000, 1500, 2000, 2500, 3000, 3500;
      29, 22.4, 17.3, 13.4, 10.4, 8];
rti = randi(5, 10, 1, 10, 10, 10, 10);

disp('Non-vectorized:')
tic
for kk = 1:rows(Lr)
  for jj = 1:length(vi)
    for ll = 1:length(vf)
      for mm = 1:length(b)
        for nn = 1:length(h)
          ratt(kk,1,jj,ll,mm,nn) = Rc(2,rti(kk,1,jj,ll,mm,nn));
        endfor
      endfor
    endfor
  endfor
endfor
toc

disp('')

disp('Vectorized:')
tic
ratt1 = Rc(2, rti(1:rows(Lr), 1, 1:length(vi), 1:length(vf), 1:length(b), 1:length(h)));
ratt1 = reshape(ratt1, [rows(Lr), 1, length(vi), length(vf), length(b), length(h)]);
toc

disp('')

if(isequal(ratt,ratt1))
  disp('ratt is equal to ratt1!')
else
  disp('ratt is not equal to ratt1!')
endif

Вывод: 

Non-vectorized:
Elapsed time is 0.16697 seconds.

Vectorized:
Elapsed time is 0.000309944 seconds.

ratt is equal to ratt1!
...