f01591a685be78db2c440e8024895601017a5b72,gpytorch/lazy/matmul_lazy_variable.py,MatmulLazyVariable,_get_indices,#MatmulLazyVariable#Any#Any#,71

Before Change

        return (left_vals.view(-1, inner_size) * right_vals.view(-1, inner_size)).sum(-1)

    def _get_indices(self, left_indices, right_indices):
        res = self.lhs.index_select(-2, left_indices) * self.rhs.index_select(-1, right_indices).transpose(-1, -2)
        return res.sum(-1)

    def diag(self):
        return (self.lhs * self.rhs.transpose(-1, -2)).sum(-1)

After Change

        right_vals = self.lhs._get_indices(_inner_repeat(inner_indices, outer_size),
                                           _outer_repeat(right_indices, inner_size))

        return (left_vals.view(-1, inner_size) * right_vals.view(-1, inner_size)).sum(-1)

    def diag(self):
        if isinstance(self.lhs, NonLazyVariable) and isinstance(self.rhs, NonLazyVariable):

Instances