For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo function
reduce_layer_norm_shape
def reduce_layer_norm_shape(input: T, gamma: T, beta: T, epsilon: Float32) -> IndexList[T.rank]
Computes the output shape for the mo.reduce.layer_norm graph op.
Args:
- input (
T): Input tensor normalized across the last dimension. - gamma (
T): Per-column scale weights applied after normalization. - beta (
T): Per-column shift weights applied after scaling. - epsilon (
Float32): Small constant added inside the normalization variance for numerical stability.
Returns:
IndexList[T.rank]: The output shape, which matches the input shape.