020d3d6dfa1bbcc13f7f7f0a833f57bb2cc8ef9d,gpytorch/lazy/constant_mul_lazy_variable.py,ConstantMulLazyVariable,__init__,#ConstantMulLazyVariable#Any#Any#,12

Before Change

    def __init__(self, lazy_var, constant):
        if not isinstance(constant, Variable):
            tensor_cls = lazy_var.tensor_cls
            constant = Variable(tensor_cls(1).fill_(constant))
        super(ConstantMulLazyVariable, self).__init__(lazy_var, constant)
        self.lazy_var = lazy_var
        self.constant = constant

After Change

class ConstantMulLazyVariable(LazyVariable):
    def __init__(self, lazy_var, constant):
        if torch.is_tensor(constant):
            constant = constant
        else:
            constant = torch.Tensor([constant], device=lazy_var.device)
        super(ConstantMulLazyVariable, self).__init__(lazy_var, constant)
        self.lazy_var = lazy_var
        self.constant = constant

Instances