Where's the implementation of KL divergence loss function?

blackbirdbarber (bbb) March 9, 2021, 11:47pm 1

Simple question but I was not able to find it due to ridiculous Github search and friends.

2 Likes

Unity05 (Unity05) March 9, 2021, 11:53pm 2

pytorch/functional.py at master · pytorch/pytorch (github.com), line 2549.

2 Likes

blackbirdbarber (bbb) March 10, 2021, 1:09pm 3

Should be in C++ I guess, else in Python.

iacob March 10, 2021, 2:59pm 4

It’s line 2627.

2 Likes

Unity05 (Unity05) March 10, 2021, 3:13pm 5

Oups, RIP my eyes then.