如何用Matlab矩阵运算代替这个循环语句。

nocturne

疑问：请看下面的代码，请问有没有办法不使用循环语句而达成相同效果，使用循环语句效率太低了。感谢！
clear
. d: k, l; I- _5 H' b& H$ sK=100;
S=500;
- R. h& B! d4 WG=rand(K,K,S);                %G为K*K*S维的矩阵
for t=1:S
: m- ]& s* C1 ~: @2 n     for m=1:K
          for n=1:K
+ H9 j  f2 A8 i6 ]8 J  I& q9 C               I=G(1:K,m,t) + G(1:K,n,t)' ;        %注意第2项的矩阵G取了转置
               E(m,n,t) = sum(I,1:2) ;
" M4 E7 q/ }" t/ I6 T0 o# z          end
     end
0 e! m( H) l$ j2 g! @6 pend
: Y  z, i2 M% D& y7 g

; e2 Y4 f- S) n" E

R_myself

你的矩阵每层都对称，利用对称性，哪怕全写循环嵌套，也可以省大约一半的计算量。
for t = 1 : S
! T, x3 ~2 R" m: m. i    for m = 1 : K
) Z( N3 x& Q0 W5 c" ]$ U          for n = m : K
) J* g) S3 r; U8 o2 ^" E7 f  U6 C. `              E2( m, n, t ) = ( sum( G( 1 : K, m, t ) ) + sum( G( 1 : K, n, t ) ) ) * K;
              E2( n, m, t ) = E2( m, n, t ); % 对称性
          end
    end
end
3 x0 T1 a9 Y1 w1 H当然更简单的方法是根据规律直接计算每层
for t = 1 : S
    M = sum( G( :, :, t ) ).' * K * ones( 1, K );
. U) G  B8 H3 O2 E    E3( :, :, t ) = M.' + M;
, R) N' p' }: i8 e9 V+ lend
在K = 50; S = 20; 时
- L( I( l4 u4 C% p0 i( X3 D原始计算方法历时 1.852707 秒。
利用对称性计算，历时 0.274421 秒。
& C, l" f: Z7 Q3 A* K! Y利用规律逐层就散，历时 0.005905 秒。

nocturne

I_believe 发表于 2023-1-6 15:12
9 i3 _( j  M  H- R/ ]3 T你这转置了，维度不一样，代码直接错误呀。

没有错误的，因为我需要做这个转置，I确实是需要这样的操作，以上循环可以运行。想问问有没有不需要用循环的方法。

I_believe

你这转置了，维度不一样，代码直接错误呀。