本项目用两种方式实现MIMO,矩阵相乘方式和torch方式
MIMO的过程为
- x
- fft
- 卷积
- ifft
- y
fft、卷积、ifft都可以用矩阵表示。
所以只要求出这三个矩阵,自然可以求出MIMO过程的矩阵H
通过 y=Hx 就可以方便地通过H求得y。
通过H的伪逆矩阵也可以方便地由y求得x。
这种方式由于由于有大矩阵相乘,实际速度偏慢,只是作为分析之用。
在噪音参数SNRdb=10的情况下,通过H的伪逆矩阵由y求得x的的score大概为0.9
其中score=1-abs(x-x_hat).mean()
其中x_为原数据,x_hat为通过y还原的x
用torch实现,避免大矩阵相乘,加快了速度