a3c1f36fe2b18b9c5193634765cf355902347c9d,gpytorch/kernels/index_kernel.py,IndexKernel,forward,#IndexKernel#Any#Any#,43

Before Change

        )

    def forward(self, i1, i2):
        covar_matrix = self.covar_factor.matmul(self.covar_factor.transpose(-1, -2))
        covar_matrix += self.log_var.exp().diag()
        covar_matrix = covar_matrix.unsqueeze(0)
        output_covar = covar_matrix.index_select(-2, i1.view(-1)).index_select(
            -1, i2.view(-1)
        )
        return output_covar

After Change

            logger.warning("Cannot infer appropriate prior from bounds. Ignoring bounds.")

    def forward(self, i1, i2):
        covar_matrix = _eval_covar_matrix(self.covar_factor, self.log_var).unsqueeze(0)
        return covar_matrix.index_select(-2, i1.view(-1)).index_select(-1, i2.view(-1))

Instances