0479ca409c32f3ed79b2f45d1b65df346390ba4b,gpytorch/lazy/lazy_variable.py,LazyVariable,__getitem__,#LazyVariable#Any#,274

Before Change

        index = list(index) if isinstance(index, tuple) else [index]
        ndimension = self.ndimension()
        index += [slice(None, None, None)] * (ndimension - len(index))
        representation = list(self.representation())

        squeeze_left = False
        squeeze_right = False
        if isinstance(index[-2], int):

After Change

                components[i] = item[batch_index]

        new_lazy_variable = self.__class__(*components)
        representation = new_lazy_variable.representation()
        ndimension = new_lazy_variable.ndimension()

        // Handle index
        left_index = index[-2]

Instances