Class PixelNorm

StyleGAN/net/model.py:18–25

#Pixel norm沿着channel维度做归一化（axis=1），这样归一化的一个好处在于，
# feature map的每个位置都具有单位长度。
class PixelNorm(torch.nn.Module):
    def __init__(self):
        super(PixelNorm, self).__init__()
        self.epsilon = 1e-8
    def forward(self,x):
        #keepdim:输出张量是否保留了dim
        out = x / torch.sqrt(torch.mean(x**2,dim = 1,keepdim=True) + self.epsilon)
        return out

#紧跟PixelNorm之后的3 x 3卷积
class WSConv2d(torch.nn.Module):