在TensorFlow 2.0中的各层之间共享权重

时间:2019-08-31 10:25:17

标签: python tensorflow keras tensorflow2.0 tensorflow-probability

我正在尝试确定在使用Tensorflow 2.0时如何在特定类型图层的多个副本之间共享权重,这些副本例如堆叠在顺序Keras模型中

特别是根据TensorFlow概率创建tfp.layers.VariationalGaussianProcess层。其输入之一是kernel_provider(您可以通过对Keras层进行子类化来实现)。

我想做的是使用一个tfp.positive_semidefinite_kernels.MaternOneHalf内核,它具有振幅和length_scale权重。然后,对于模型中的所有VariationalGaussianProcess层,要在所有这些层之间共享振幅和length_scale权重。

0 个答案:

没有答案