我试图确定在使用 Tensorflow 2.0 时如何在特定类型层的多个副本之间共享权重,例如堆叠在顺序 Keras 模型中
具体从TensorFlow概率一个tfp.layers.VariationalGaussianProcess
层。它的一个输入是 a kernel_provider
(您通过对 Keras 层进行子类化来制作)。
我想做的是使用tfp.positive_semidefinite_kernels.MaternOneHalf
具有振幅和length_scale
权重的内核。然后对于模型中的所有层,在所有这些层之间共享VariationalGaussianProcess
幅度和权重。length_scale