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

Before Change

                .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)
            factor = lazy_var._t_matmul(res).permute(2, 1, 0)

After Change

            n_batch = res.size(0)
            res = res.view(n_batch, lazy_var.size(-2), -1)
            factor = lazy_var._t_matmul(res)
            factor = factor.view(n_batch, lazy_var.size(-1), -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(-2), -1)

Instances