xlnstorch.operators.lns_layer_norm

xlnstorch.operators.lns_layer_norm(x, normalized_shape, base, weight=None, bias=None, eps=1e-05)

See docs for torch.nn.functional.layer_norm() for parameter/return details. Typically, torch.Tensor arguments are the equivalent internal representations of LNStensors.

Parameters:

• x (Tensor)

• normalized_shape (int | Tuple[int])

• base (Tensor)

• weight (Tensor | None)

• bias (Tensor | None)

• eps (Tensor)

Return type:

Tensor