是否可以修正tfpl.IndependentNormal中的平均值/var参数？

Question

我用Python编写了一个标准的概率神经网络，最后一层是tfp.layers.IndependentNormal，这给了我一个正态分布。然而，我只想训练上述分布的平均值，保持方差不变

有没有人尝试过类似的方法，或者知道如何做到这一点

Answer 1

您可以这样做：

model = Sequential([
    ...
    tfpl.DistributionLambda(lambda t: tfd.Independent(tfd.Normal(loc = t, scale = 0.5)))
])

在这里，您需要将scale设置为一个常量值以保持固定