Initialize LayerNorm. Args: *args: Variable positional arguments. **kwargs: Additional keyword arguments.
(self, *args, **kwargs)
| 299 | |
| 300 | class LayerNorm(nn.LayerNorm): |
| 301 | def __init__(self, *args, **kwargs): |
| 302 | """Initialize LayerNorm. |
| 303 | |
| 304 | Args: |
| 305 | *args: Variable positional arguments. |
| 306 | **kwargs: Additional keyword arguments. |
| 307 | """ |
| 308 | super().__init__(*args, **kwargs) |
| 309 | |
| 310 | def forward(self, input): |
| 311 | """Forward pass for training. |