xlnstorch.operators.lns_mse_loss

xlnstorch.operators.lns_mse_loss(x, y, base, size_average=None, reduce=None, reduction='mean', weight=None)

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

Parameters:

• x (Tensor)

• y (Tensor)

• base (Tensor)

• size_average (bool)

• reduce (bool)

• reduction (str)

• weight (Tensor)

Return type:

Tensor