Module: tfp.layers.weight_norm

View source on GitHub

Layer wrapper for weight normalization.

Classes

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