xlnstorch.operators.lns_batch_norm

xlnstorch.operators.lns_batch_norm(x, running_mean, running_var, momentum, eps, base, weight=None, bias=None, training=False)

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

Parameters:

• x (Tensor)

• running_mean (Tensor)

• running_var (Tensor)

• momentum (Tensor)

• eps (Tensor)

• base (Tensor)

• weight (Tensor | None)

• bias (Tensor | None)

• training (bool)

Return type:

Tensor