fix: typo

This commit is contained in:
gehbiszumeis 2021-10-27 16:06:04 +02:00
parent 4de3fdda20
commit f79144bf47

View file

@ -106,7 +106,7 @@ class VariationalPosterior(Parameterized):
self.link_parameters(self.mean, self.variance)
self.num_data, self.input_dim = self.mean.shape
if self.has_uncertain_inputs():
assert self.variance.shape == self.mean.shape, "need one variance per sample and dimenion"
assert self.variance.shape == self.mean.shape, "need one variance per sample and dimension"
def set_gradients(self, grad):
self.mean.gradient, self.variance.gradient = grad