Tf 约束可训练变量

比如现在有一个公式,公式中的n个β是系数,我想把他设置成learnable的 ,但是他有一个约束是β必须大于0且所有β求和等以一个常数,这种约束在tensorflow中应该怎么满足。
我想通过tf.layer.dense构建这组系数,如果dense的时候可以约束最好了。@snowkylin

看起来是softmax的一个扩展,你可以让beta直接经过一个softmax层(所有beta大于0且求和等于1)再乘上你想要的常数。

关于对可训练常量的约束可以参考下面的一些例子

1 Like