Fix error in old PyTorch for KL and JS divergence

[kzkadc]

Fixes https://github.com/pytorch/ignite/pull/3232#issuecomment-2063322746

Check list:

• [ ] New tests are added (if a new feature is added)
• [ ] New doc strings: description and/or example code are in RST format
• [ ] Documentation is updated (if required)

[vfdev-5]

There are unrelated CI failures to fix later, so will be merging this PR.