60a342edc8b501802135df44869353cc8604d838,gpytorch/lazy/kronecker_product_lazy_variable.py,KroneckerProductLazyVariable,_matmul_closure_factory,#KroneckerProductLazyVariable#,31

Before Change

    def _matmul_closure_factory(self, *args):
        if len(args) == 1:
            columns, = args

            def closure(mat2):
                return sym_kronecker_product_toeplitz_matmul(columns, mat2)

After Change

        sub_matmul_closures = []
        i = 0
        for lazy_var in self.lazy_vars:
            len_repr = len(lazy_var.representation())
            sub_matmul_closure = lazy_var._matmul_closure_factory(*args[i:i + len_repr])
            sub_matmul_closures.append(sub_matmul_closure)
            i = i + len_repr

        def closure(tensor):
            is_vec = tensor.ndimension() == 1
            if is_vec:

Instances