def __init__(self, n_antenna=64, n_beam=64, theta=None):
super(AnalogBeamformer, self).__init__()
self.codebook = PhaseShifter(in_features=2 * n_antenna,
out_features=n_beam,
scale=np.sqrt(n_antenna),
theta=theta)
self.beam_selection = PowerPooling(2 * n_beam)
def __init__(self, n_antenna, n_beam):
super(AnalogBeamformer, self).__init__()
self.codebook = PhaseShifter(in_features=2*n_antenna, out_features=n_beam, scale=np.sqrt(n_antenna))
self.beam_selection = PowerPooling(2*n_beam)
self.compute_power = ComputePower(2*n_beam)