Format of weight parameters in KLDivLoss

KLDivLoss can take a weight parameter but the docs don’t specify how it should be formatted. What should the format be?

It should be a 1D tensor having as many elements as you have classes. We’ll have to add that to the docs, thanks for reporting that!

Can it? I can see nothing weight related in the docs, nor in the source.