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

Before Change

        else:
            res = res.t().contiguous().view(n_cols, lazy_var.size(-1), -1)
            factor = lazy_var._matmul(res)
            if factor.ndimension() == 3:
                factor = factor.permute(2, 1, 0)
            else:
                factor = factor.transpose(-2, -1)
            res = factor.contiguous().view(-1, n_cols)
    return res

After Change

            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