“Support size” refers to the dimension of the probability vector.
For example, if x, y are of dimension B x D (batch size = B, support size = D),
F.kl_div(x, y, reduction='mean') * D is equal to (F.kl_div(x, y, reduction='batchmean')