1b809b3c9570d617cb26949e40b921b36e826580,gpytorch/lazy/kronecker_product_lazy_variable.py,,_matmul,#Any#Any#,17

Before Change

        if is_batch:
            n_batch = res.size(0)
            res = (
                res.transpose(-2, -1)
                .contiguous()
                .view(n_batch, n_cols, lazy_var.size(-1), -1)
                .transpose(, )
                .contiguous()
            )
            factor = lazy_var._matmul(res).permute(1, 3, 2, 0)

After Change

    for lazy_var in list(lazy_vars)[::-1]:
        if is_batch:
            n_batch = res.size(0)
            res = res.view(n_batch, lazy_var.size(-1), -1)
            factor = lazy_var._matmul(res)
            factor = factor.view(n_batch, lazy_var.size(-2), -1, n_cols).transpose(-3, -2).contiguous().view(-1, n_cols)
            res = factor.contiguous().view(n_batch, -1, n_cols)
        else:
            res = res.view(lazy_var.size(-1), -1)
            factor = lazy_var._matmul(res)

Instances