979b8c9efa551e8c948a4aca145367a2d87ac8d6,gpytorch/distributions/multivariate_normal.py,_MultivariateNormalBase,expand,#_MultivariateNormalBase#Any#,63

Before Change

            self.__unbroadcasted_scale_tril = ust

    def expand(self, batch_size):
        new_loc = self.loc.expand(torch.Size(list(batch_size) + [self.loc.size(-1)]))
        new_covar = self._covar.expand(torch.Size(list(batch_size) + list(self._covar.shape[-2:])))
        res = self.__class__(new_loc, new_covar)
        return res

After Change

    def expand(self, batch_size):

        new_loc = self.loc.expand(torch.Size(batch_size) + self.loc.shape[-1:])
        new_covar = self._covar.expand(torch.Size(batch_size) + self._covar.shape[-2:])
        res = self.__class__(new_loc, new_covar)
        return res

Instances