更新:此问题已由https://github.com/GPflow/GPflow/pull/1594解决,它将成为下一个 GPflow 补丁版本 (2.1.4) 的一部分。
通过该修复,您不需要自定义类。您需要做的就是None
沿第一个维度显式设置静态形状:
inducing_variable = gpflow.inducing_variables.InducingPoints(
tf.Variable(
Z1, # initial value
trainable=False, # True does not work - see Note below
shape=(None, Z1.shape[1]), # or even tf.TensorShape(None)
dtype=gpflow.default_float(), # required due to tf's 32bit default
)
)
m = gpflow.models.SGPR(data=(X, Y), kernel=k, inducing_variable=inducing_variable)
然后m.inducing_variable.Z.assign(Z2)
应该可以正常工作。
请注意,在这种情况下Z
无法训练,因为 TensorFlow 优化器需要在构建时知道形状并且不支持动态形状。
现在(从 GPflow 2.1.2 开始)没有内置方法可以改变 的诱导变量的形状SGPR
,尽管原则上是可行的。你可以用你自己的诱导变量类得到你想要的东西:
class VariableInducingPoints(gpflow.inducing_variables.InducingPoints):
def __init__(self, Z, name=None):
super().__init__(Z, name=name)
# overwrite with Variable with None as first element in shape so
# we can assign arrays with arbitrary length along this dimension:
self.Z = tf.Variable(Z, dtype=gpflow.default_float(),
shape=(None, Z.shape[1])
)
def __len__(self):
return tf.shape(self.Z)[0] # dynamic shape
# instead of the static shape returned by the InducingPoints parent class
然后做
m = gpflow.models.SGPR(
data=(X, Y), kernel=k, inducing_variable=VariableInducingPoints(Z1)
)
反而。然后你m.inducing_variable.Z.assign()
应该随心所欲地工作。
(对于SVGP
,诱导变量的大小和由q_mu
和定义的分布q_sqrt
必须匹配,并且在构造时就知道了,所以在这种情况下,改变诱导变量的数量就不那么简单了。)