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

Detailed Description

Definition at line 9 of file spectral_norm.py.

---

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

• torch/nn/utils/spectral_norm.py