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
	name
	dim

Detailed Description

Definition at line 8 of file weight_norm.py.

---

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

• torch/nn/utils/weight_norm.py