1b809b3c9570d617cb26949e40b921b36e826580,gpytorch/lazy/kronecker_product_lazy_variable.py,,_t_matmul,#Any#Any#,44

Before Change

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

After Change

        else:
            res = res.view(lazy_var.size(-2), -1)
            factor = lazy_var._t_matmul(res)
            factor = factor.view(lazy_var.size(-1), -1, n_cols).transpose(-3, -2).contiguous().view(-1, n_cols)
            res = factor.contiguous().view(-1, n_cols)
    return res

Instances