如何在张量流中实现负二项式可能罩函数并将其用作损失函数来训练RNN？

Question

我正在尝试预测循环神经网络每一步中随机变量的均值和方差。我需要通过使用负二项式对数似然作为损失函数来训练该模型，其中模型学习输出均值和方差以及定义分布的每个时间步。有人可以帮我在Tensorflow中实现此损失功能吗？