Skip to main content
Log in

Python module

layer_norm

Layer Normalization layer.

LayerNorm

class max.nn.norm.layer_norm.LayerNorm(dims, device, dtype, eps=1e-05, use_bias=True)

Layer normalization block.

Parameters:

LayerNormV1

class max.nn.norm.layer_norm.LayerNormV1(weight, bias=None, eps=1e-06)

Layer normalization block.

Deprecated: Use LayerNorm instead.

Parameters:

bias

bias*: TensorValue | None* = None

eps

eps*: float* = 1e-06

weight

weight*: TensorValue*