def distribution(self, distr_args, loc=None, scale=None, dim=None):
dist = LowrankMultivariateGaussian(dim, self.rank, *distr_args)
if loc is None and scale is None:
return dist
else:
return TransformedDistribution(
dist, [bijection.AffineTransformation(loc=loc, scale=scale)])
def distribution(self, distr_args, scale=None, **kwargs) -> Distribution:
# todo dirty way of calling for now, this can be cleaned
distr = LowrankMultivariateGaussian(self.dim, self.rank, *distr_args)
if scale is None:
return distr
else:
return TransformedDistribution(
distr, bijection.AffineTransformation(scale=scale))
def distribution(self, distr_args, scale=None, **kwargs) -> Distribution:
distr = LowrankMultivariateGaussian(self.dim, self.rank, *distr_args)
if scale is None:
return distr
else:
return TransformedDistribution(
distr, [bijection.AffineTransformation(scale=scale)]
)