是否可以修复 tfpl.IndependentNormal 中的均值/var arg？

Question

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

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

Answer 1

你可以这样做：

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

这里你需要将scale设置成一个常数值来保持它的固定。