Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.nn.modules.normalization.LayerNorm Class Reference
Inheritance diagram for torch.nn.modules.normalization.LayerNorm:

Public Member Functions

def __init__ (self, normalized_shape, eps=1e-5, elementwise_affine=True)
 
def reset_parameters (self)
 
def forward (self, input)
 
def extra_repr (self)
 

Public Attributes

 normalized_shape
 
 eps
 
 elementwise_affine
 
 weight
 
 bias
 

Detailed Description

Definition at line 76 of file normalization.py.


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