How to speed up this function?

2 次查看（过去 30 天）
Jonathan Mayers
Jonathan Mayers 2016-5-26
回答： Jonathan Mayers 2016-5-26
Hi all,
I am calculating a value Gamma as shown in the image below.
I have implemented this calculation as a for-loop and I suspect it could be done without using a for-loop and be faster. Every h term is the jth column of an NxK matrix and Pj is the jth column of KxK diagonal matrix P. What is the fastest way to calculate this value?
function [ GAMMA ] = gamma_dnc( P,Hh,Ht,K )
%GAMMA_DNC Calculates the component GAMMA in the DNC
% beamforming matrix V.
% Running total
total = 0;
for j = 1:K
p = P(j,j); % Select jth column of P
ht = Ht(:,j); % Select jth column of Ht
hh = Hh(:,j); % Select jth column of Hh
% Perform summation
h1 = ht*(hh');
h2 = hh*(ht');
h3 = ht*(ht');
h4 = h1 + h2 + h3;
total = total + p*h4;
end
GAMMA = mean2(total);
end
  1 个评论
James Tursa
James Tursa 2016-5-26
What are the dimensions of the variables? And which variables are complex?

请先登录，再进行评论。

请先登录，再回答此问题。

采纳的回答

Jonathan Mayers
Jonathan Mayers 2016-5-26
Matrix P is a real-valued matrix and the H matrices are complex-valued matrices. I have found a way to do it without looping.
function [ GAMMA ] = gamma_dnc( P,Hh,Ht )
%GAMMA_DNC Calculates the component GAMMA in the DNC
% beamforming matrix V.
% Store transposes of matrices to reduce
% function call overhead
Hh_tp = Hh';
Ht_tp = Ht';
% Compute intermediate products
H1 = Ht*Hh_tp;
H2 = Hh*Ht_tp;
H3 = Ht*Ht_tp;
H4 = H1 + H2 + H3;
H5 = P(1,1)*H4; % All users transmit with equal power
T = sum(H5); % Find sum of all columns
GAMMA = mean(T)/10; % Have to divide by 10 for some reason
end

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Operating on Diagonal Matrices 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by