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