38a40828ec688be626e51ebcb1a60a54479fd997,gpytorch/lazy/kronecker_product_lazy_variable.py,KroneckerProductLazyVariable,mul,#KroneckerProductLazyVariable#Any#,161

Before Change

        Returns:
            - KroneckerProductLazyVariable with columns[0] = columns[0]*(constant) and columns[i] = columns[i] for i>0
        
        columns = self.columns.clone()
        columns[0] = columns[0].mul(constant)
        return KroneckerProductLazyVariable(columns, self.J_lefts, self.C_lefts,
                                            self.J_rights, self.C_rights, self.added_diag)

    def mul_(self, constant):

After Change

            - KroneckerProductLazyVariable with columns[0] = columns[0]*(constant) and columns[i] = columns[i] for i>0
        
        columns = self.columns
        constant_tensor = torch.zeros(columns.size()) + 1
        constant_tensor[0] = constant_tensor[0] * constant.data
        constant_variable = Variable(constant_tensor)
        columns = columns * constant_variable
        return KroneckerProductLazyVariable(columns, self.J_lefts, self.C_lefts,
                                            self.J_rights, self.C_rights, self.added_diag)

Instances