-
Notifications
You must be signed in to change notification settings - Fork 1
/
getM.m
42 lines (27 loc) · 1.09 KB
/
getM.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
function [M] = getM(Y,D,X,k)
%Update the M (E_R*E_R') across all sites
% Input: Y - Signals across 'N' sites ( m x n x N )
% D - Dictionary across 'N' sites ( m x K x N )
% X - sparse coding ( K x n x N )
% hat and tilda should be same dimension
% k - kth atom we are updating for each site's local dictionary
% Output: M - Error matrix across 'N' sites ( m x n x N )
[m,n,N] = size(Y);
%% Create M matrix (m x m x N ) & Cell Arrays
M = zeros(m,m,N);
%%
for i=1:N % for each site node...
Ind_nz = find(X(k,:,i)); %find all non-zero indexs, X_t if needed
nz = length(Ind_nz);
if(nz~=0)
Omega = zeros(n,nz);
for col=1:nz % for each column in Omega
Omega(Ind_nz(col),col) = 1;
end
P = D(:,:,i)*X(:,:,i) - D(:,k,i)*X(k,:,i);
E = Y(:,:,i) - P;
E_R = E*Omega;
M(:,:,i) = E_R*(E_R');
end
end
end