Google I/O returns May 18-20! Reserve space and build your schedule Register now

Module: tfp.layers.weight_norm

Layer wrapper for weight normalization.

Classes

class WeightNorm: Layer wrapper to decouple magnitude and direction of the layer's weights.