Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
torch.nn.utils.weight_norm.WeightNorm Class Reference
Inheritance diagram for torch.nn.utils.weight_norm.WeightNorm:

Public Member Functions

def __init__ (self, name, dim)
def compute_weight (self, module)
def remove (self, module)
def __call__ (self, module, inputs)

Static Public Member Functions

def apply (module, name, dim)

Public Attributes


Detailed Description

Definition at line 8 of file

The documentation for this class was generated from the following file: