tensorplay.nn.functional.kl_div

tensorplay.nn.functional.kl_div(input: Tensor, target: Tensor, size_average=None, reduce=None, reduction: str = 'mean', log_target: bool = False) → Tensor[source]

Compute the KL Divergence loss.

input holds log-probabilities; see KLDivLoss. Note that reduction='mean' divides by the number of elements and does not return the true KL divergence value — use 'batchmean'.