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.spectral_norm.SpectralNorm Class Reference
Inheritance diagram for torch.nn.utils.spectral_norm.SpectralNorm:

Public Member Functions

def __init__ (self, name='weight', n_power_iterations=1, dim=0, eps=1e-12)
def reshape_weight_to_matrix (self, weight)
def compute_weight (self, module, do_power_iteration)
def remove (self, module)
def __call__ (self, module, inputs)

Static Public Member Functions

def apply (module, name, n_power_iterations, dim, eps)

Public Attributes


Detailed Description

Definition at line 9 of file

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