33368888efbdeaf2402eb52d14f1c4ea0c853909,gpytorch/lazy/root_lazy_variable.py,RootLazyVariable,_batch_get_indices,#RootLazyVariable#Any#Any#Any#,63

Before Change

            left_vals = self.root._batch_get_indices(
                _outer_repeat(batch_indices, inner_size),
                _outer_repeat(left_indices, inner_size),
                _inner_repeat(inner_indices, outer_size),
            )
            right_vals = self.root.transpose(-1, -2)._batch_get_indices(
                _outer_repeat(batch_indices, inner_size),
                _inner_repeat(inner_indices, outer_size),
                _outer_repeat(right_indices, inner_size),
            )

After Change

            batch_indices = _outer_repeat(batch_indices, inner_size)
            left_indices = _outer_repeat(left_indices, inner_size)
            right_indices = _outer_repeat(right_indices, inner_size)
            inner_indices = _inner_repeat(inner_indices, outer_size)

            left_vals = self.root._batch_get_indices(
                batch_indices, left_indices, inner_indices
            )

Instances