33368888efbdeaf2402eb52d14f1c4ea0c853909,gpytorch/lazy/matmul_lazy_variable.py,MatmulLazyVariable,_batch_get_indices,#MatmulLazyVariable#Any#Any#Any#,67

Before Change

            torch.arange(0, inner_size, out=inner_indices.data)

            left_vals = self.lhs._batch_get_indices(
                _outer_repeat(batch_indices, inner_size),
                _outer_repeat(left_indices, inner_size),
                _inner_repeat(inner_indices, outer_size),
            )

After Change

            torch.arange(0, inner_size, out=inner_indices.data)

            // Repeat the indices to get all the appropriate terms
            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)

Instances