2012-11-15 5 views
9

Ich habe Probleme beim Visualisieren, wie diese Schleifengruppe vektorisiert wird. Jede Anleitung würde geschätzt werden.So verschachteln Sie eine verschachtelte Schleife

ind_1 = [1,2,3]; 
ind_2 = [1,2,4]; 
K = zeros(3,3,3,3,3,3,3,3,3); 
pp = rand(4,4,4); 

for s = 1:3 
for t = 1:3 
    for k = 1:3 
    for l = 1:3 
    for m = 1:3 
    for n = 1:3 
     for o = 1:3 
     for p = 1:3 
     for r = 1:3 
     % the following loops are singular valued except when 
     % y=3 for ind_x(y) in this case 
     for a_s = ind_1(s):ind_2(s) 
      for a_t = ind_1(t):ind_2(t) 
      for a_k = ind_1(k):ind_2(k) 
      for a_l = ind_1(l):ind_2(l) 
      for a_m = ind_1(m):ind_2(m) 
       for a_n = ind_1(n):ind_2(n) 
       for a_o = ind_1(o):ind_2(o) 
       for a_p = ind_1(p):ind_2(p) 
       for a_r = ind_1(r):ind_2(r) 
        K(s,t,k,l,m,n,o,p,r) = K(s,t,k,l,m,n,o,p,r) + ... 
        pp(a_s, a_t, a_r) * pp(a_k, a_l, a_r) * ... 
        pp(a_n, a_m, a_s) * pp(a_o, a_n, a_t) * ... 
        pp(a_p, a_o, a_k) * pp(a_m, a_p, a_l); 
       end 
       end 
       end 
       end 
      end 
      end 
      end 
      end 
     end 
     end 
     end 
     end 
    end 
    end 
    end 
    end 
end 
end 

EDIT:

Der Code wird durch Summieren der Werte eines Produkts von pp s ein oder zwei Mal für jeden Index, in Abhängigkeit vom Wert einen Rang-9-Tensor mit Indizes von 1 bis 3 Schaffung von ind_1 und ind_2.

EDIT:

Hier ist ein 3D-Beispiel, obwohl bedenken, dass die Tatsache, dass die Indizes von pp sind einfach in der 9d Version nicht erhalten permutiert ist:

ind_1 = [1,2,3]; 
ind_2 = [1,2,4]; 
K = zeros(3,3,3); 
pp = rand(4,4,4); 

for s = 1:3 
for t = 1:3 
    for k = 1:3 
    % the following loops are singular valued except when 
    % y=3 for ind_x(y) in this case 
    for a_s = ind_1(s):ind_2(s) 
    for a_t = ind_1(t):ind_2(t) 
    for a_k = ind_1(k):ind_2(k) 
     K(s,t,k) = K(s,t,k) + ... 
     pp(a_s, a_t, a_r) * pp(a_t, a_s, a_k) * ... 
     pp(a_k, a_t, a_s) * pp(a_k, a_s, a_t); 
    end 
    end 
    end 
    end 
end 
end 
+0

Können Sie ein wenig über das, was dieser Code tut, ausarbeiten und Kommentare hinzufügen? – igon

+0

@igon: Ich habe bearbeitet. – erbridge

+0

Können Sie ein 2D- oder 3D-Beispiel zur Veranschaulichung erstellen? Es wäre einfacher für uns, damit zu arbeiten, und das Erstellen des Beispiels könnte Ihnen helfen, eine Strategie selbst zu finden. – tmpearce

Antwort

5

Woh! Ziemlich einfache Lösung, war aber nicht leicht zu finden. Übrigens frage ich mich, woher deine Formel kommt.

Wenn Sie nicht vorübergehend nichts dagegen ein bisschen einen Speicher (2 mal 4^9 Arrays vs 3^9 vorher) zu verlieren, können Sie Ansammlung von 3. und 4. Hyperebenen am Ende verschieben.

Testing mit Oktave 3.2.4 auf einem Unix-Rechner, fällt es aus 23s (67MB) zu 0.17s (98Mb).

function K = tensor9_opt(pp) 

    ppp = repmat(pp, [1 1 1 4 4 4 4 4 4]) ; 
    % The 3 first numbers are variable indices (eg 1 for a_s to 9 for a_r) 
    % Other numbers must complete 1:9 indices in any order 
    T = ipermute(ppp, [1 2 9 3 4 5 6 7 8]) .* ... 
     ipermute(ppp, [3 4 9 1 2 5 6 7 8]) .* ... 
     ipermute(ppp, [6 5 1 2 3 4 7 8 9]) .* ... 
     ipermute(ppp, [7 6 2 1 3 4 5 8 9]) .* ... 
     ipermute(ppp, [8 7 3 1 2 4 5 6 9]) .* ... 
     ipermute(ppp, [5 8 4 1 2 3 6 7 9]) ; 

    % I have not found how to manipulate 'multi-ranges' programmatically. 
    T1 = T (:,:,:,:,:,:,:,:,1:end-1) ; T1(:,:,:,:,:,:,:,:,end) += T (:,:,:,:,:,:,:,:,end) ; 
    T = T1(:,:,:,:,:,:,:,1:end-1,:) ; T (:,:,:,:,:,:,:,end,:) += T1(:,:,:,:,:,:,:,end,:) ; 
    T1 = T (:,:,:,:,:,:,1:end-1,:,:) ; T1(:,:,:,:,:,:,end,:,:) += T (:,:,:,:,:,:,end,:,:) ; 
    T = T1(:,:,:,:,:,1:end-1,:,:,:) ; T (:,:,:,:,:,end,:,:,:) += T1(:,:,:,:,:,end,:,:,:) ; 
    T1 = T (:,:,:,:,1:end-1,:,:,:,:) ; T1(:,:,:,:,end,:,:,:,:) += T (:,:,:,:,end,:,:,:,:) ; 
    T = T1(:,:,:,1:end-1,:,:,:,:,:) ; T (:,:,:,end,:,:,:,:,:) += T1(:,:,:,end,:,:,:,:,:) ; 
    T1 = T (:,:,1:end-1,:,:,:,:,:,:) ; T1(:,:,end,:,:,:,:,:,:) += T (:,:,end,:,:,:,:,:,:) ; 
    T = T1(:,1:end-1,:,:,:,:,:,:,:) ; T (:,end,:,:,:,:,:,:,:) += T1(:,end,:,:,:,:,:,:,:) ; 
    K = T (1:end-1,:,:,:,:,:,:,:,:) ; K (end,:,:,:,:,:,:,:,:) += T (end,:,:,:,:,:,:,:,:) ; 
endfunction 

pp = rand(4,4,4); 
K = tensor9_opt(pp) ; 
+0

Ausgezeichnet. Vielen Dank. – erbridge