torch.nn.modules.normalization.GroupNorm Class Reference

Inheritance diagram for torch.nn.modules.normalization.GroupNorm:

Public Member Functions
def	__init__ (self, num_groups, num_channels, eps=1e-5, affine=True)
def	reset_parameters (self)
def	forward (self, input)
def	extra_repr (self)

Public Attributes
	num_groups
	num_channels
	eps
	affine
	weight
	bias

Detailed Description

Definition at line 166 of file normalization.py.

---

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

• torch/nn/modules/normalization.py