Takes softmax on both sides and returns KL divergence Note: - Returns the sum over all examples. Divide by the batch size afterwards if you want the mean. - Sends gradients to inputs but not the targets.
(input_logits, target_logits, sigmoid=False)
source not stored for this graph (policy: none)