Method __init__

StyleGAN/net/model.py:19–21

(self)

# 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)