torch.nn.modules.batchnorm._BatchNorm Class Reference

Inheritance diagram for torch.nn.modules.batchnorm._BatchNorm:

Public Member Functions
def	__init__ (self, num_features, eps=1e-5, momentum=0.1, affine=True, track_running_stats=True)
def	reset_running_stats (self)
def	reset_parameters (self)
def	forward (self, input)
def	extra_repr (self)

Public Attributes
	num_features
	eps
	momentum
	affine
	track_running_stats
	weight
	bias

Detailed Description

Definition at line 15 of file batchnorm.py.

---

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

• torch/nn/modules/batchnorm.py