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.
LayerNormV1
class max.nn.norm.layer_norm.LayerNormV1(weight, bias=None, eps=1e-06)
Layer normalization block.
Deprecated: Use LayerNorm instead.
-
Parameters:
-
- weight (
TensorValue
) - bias (
TensorValue
|
None
) - eps (
float
)
- weight (
bias
bias*: TensorValue | None* = None
eps
eps*: float* = 1e-06
weight
weight*: TensorValue*
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!