c517e0b8b34a85b8142b4669c152b6e62c02d8e2,gpytorch/lazy/lazy_tensor.py,LazyTensor,expand,#LazyTensor#,599

Before Change

            raise RuntimeError("Invalid arguments {} to expand.".format(sizes))

        current_shape = torch.Size([1 for _ in range(len(shape) - self.dim())] + list(self.shape))
        repeat_shape = torch.Size(
            [expand_size // current_size for expand_size, current_size in zip(shape, current_shape)]
        )
        return self.repeat(*repeat_shape)

    @cached
    def evaluate(self):

After Change

        else:
            raise RuntimeError("Invalid arguments {} to expand.".format(sizes))

        res = self._expand_batch(batch_shape=shape[:-2])
        return res

    @cached
    def evaluate(self):

Instances