信息论-matlab求信道容量（迭代法）

3-9解答

%%以习题3-3中信道为例，求解其信道容量

clc;
clear all;
P_YX=[0.98 0.02;
      0.02 0.98];%信道的转移概率矩阵
e=0.000001;%停止迭代的容差限
C1=1;C=0;N=0;
r=size(P_YX);
P_X=ones(1,r(1))/r(1);
%迭代求解
while (abs(C1-C))>e
    P_Y=P_X*P_YX;
    I1=sum((P_YX.*log2(P_YX))');
    I2=log2(P_Y)*(P_YX');
    BETA=exp(I1-I2);
    B=P_X*(BETA');
    C1=log(B);C=log(max(BETA));
    P_X=P_X.*BETA/B;
    N=N+1;
end;
disp(['信道容量为：',num2str(C)]);
disp(['迭代次数为：',num2str(N)]);
%%  ----运行结果---
% 信道容量为：0.85856
% 迭代次数为：1