f01591a685be78db2c440e8024895601017a5b72,gpytorch/lazy/matmul_lazy_variable.py,MatmulLazyVariable,_batch_get_indices,#MatmulLazyVariable#Any#Any#Any#,55

Before Change

        right_indices = right_indices.data

        inner_size = self.lhs.size(-1)
        inner_indices = right_indices.new(inner_size)
        torch.arange(0, inner_size, out=inner_indices)

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

After Change

    def _batch_get_indices(self, batch_indices, left_indices, right_indices):
        outer_size = batch_indices.size(0)
        inner_size = self.lhs.size(-1)
        inner_indices = Variable(right_indices.data.new(inner_size))
        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),

Instances