是否可以修复 tfpl.IndependentNormal 中的 mean/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设置成一个常量来保持固定。