I'm not sure I quite understand the issue here, but if helps I've done a little derivation for the divergence loss formula here which might clear it up: https://drive.google.com/file/d/1PdGRB8k0okYGeEUZTTmlBQ1MKGqb5mEh/view?usp=sharing