Normal gaussian N(0, 1) noise

def gaussian_noise(in_features, out_features, device):
    """ Normal gaussian N(0, 1) noise """
    return torch.randn((in_features, out_features), device=device), torch.randn(out_features, device=device)