# Copyright (c) IMToolkit Development Team
# This toolkit is released under the MIT License, see LICENSE.txt
from .Channel import Channel
from .Util import xp, randn_c
[docs]class AWGNChannel(Channel):
"""
A `Channel` class that generates the additive white Gaussian noise (AWGN). All the channel matrix is set to an identity matrix.
"""
def __init__(self, IT, M):
"""
Args:
IT (int): the number of parallel channel matrices.
M (int): the number of transmit antennas or subcarriers.
"""
self.IT = IT
self.M = M
self.N = M
[docs] def randomize(self):
self.channelMatrix = xp.tile(xp.eye(self.M), self.IT).T
[docs] def getChannel(self):
return self.channelMatrix
[docs] def getEstimate(self):
return self.channelMatrix
