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