Как вычислить отрицательную потерю логарифмической вероятности в тензорном потоке 2.1.0?

Question

Мне нужно реализовать отрицательную логарифмическую потерю вероятности распределения Гаусса (я хочу оценить как среднее значение, так и дисперсию распределения) в Tensorflow 2.1.0. Какой самый элегантный способ сделать это при использовании API подкласса модели?