import torch import torch.nn as nn def AWGN_channel(): pass def Rayleigh_channel(): pass