91b0d220c8e816766fd4565e1d2f5115d3afbefe,gpytorch/lazy/lazy_tensor.py,LazyTensor,evaluate,#LazyTensor#,408

Before Change

        if num_rows < num_cols:
            eye = torch.eye(num_rows, dtype=self.dtype, device=self.device)
            if batch_mode:
                eye = eye.unsqueeze(0).expand(batch_size, num_rows, num_rows)
                return self.transpose(1, 2).matmul(eye).transpose(1, 2).contiguous()
            else:
                return self.t().matmul(eye).t().contiguous()
        else:
            eye = torch.eye(num_cols, dtype=self.dtype, device=self.device)
            if batch_mode:
                eye = eye.unsqueeze(0).expand(batch_size, num_cols, num_cols)
            return self.matmul(eye)

    def evaluate_kernel(self):
        

After Change

        if num_rows < num_cols:
            eye = torch.eye(num_rows, dtype=self.dtype, device=self.device)
            eye = eye.expand(*self.batch_shape, num_rows, num_rows)
            return self.transpose(-1, -2).matmul(eye).transpose(-1, -2).contiguous()
        else:
            eye = torch.eye(num_cols, dtype=self.dtype, device=self.device)
            eye = eye.expand(*self.batch_shape, num_cols, num_cols)

Instances