1972461c3c06d11872cdf62941a616ee82c858a8,gpytorch/lazy/constant_mul_lazy_variable.py,ConstantMulLazyVariable,_constant_as,#ConstantMulLazyVariable#Any#,46

Before Change

        if constant.ndimension() > other.ndimension():
            constant = constant.squeeze(-1)

        return constant.expand_as(other)

    def _size(self):
        return self.lazy_var.size()

After Change

        return res

    def _constant_as(self, other):
        size = [self.constant.numel()] + [1] * (other.ndimension() - 1)
        constant = self.constant.view(*size)

        if constant.ndimension() > other.ndimension():
            constant = constant.squeeze(-1)

        return constant

    def _size(self):
        return self.lazy_var.size()

Instances