xlnstorch.operators.lns_kl_div

xlnstorch.operators.lns_kl_div(x, y, base, size_average=None, reduce=None, reduction='mean', log_target=False)

See docs for torch.nn.functional.kl_div() 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)

• log_target (bool)

Return type:

Tensor