def remove(module): KernelNorm.remove(module, SpectralNorm.NAME)
def remove(module): KernelNorm.remove(module, WeightNorm.NAME)
def add(module, norm=1): KernelNorm.add(module, SpectralNorm(norm))
def add(module, scale=True, axis=-1, epsilon=1e-5): KernelNorm.add(module, WeightNorm(scale, axis, epsilon))