a361f45c09f173f662141ac42d4c6c3b8c304be8,gpytorch/lazy/sum_lazy_tensor.py,SumLazyTensor,__init__,#SumLazyTensor#,14

Before Change

class SumLazyTensor(LazyTensor):
    def __init__(self, *lazy_tensors, **kwargs):
        lazy_tensors = list(lazy_tensors)
        for i, lazy_tensor in enumerate(lazy_tensors):
            try:
                lazy_tensors[i] = lazify(lazy_tensor)
            except TypeError:
                raise TypeError("All arguments of a SumLazyTensor should be LazyTensors or Tensors")
        super(SumLazyTensor, self).__init__(*lazy_tensors, **kwargs)

        self.lazy_tensors = lazy_tensors

After Change

        except TypeError:
            raise TypeError("All arguments of a SumLazyTensor should be LazyTensors or Tensors")
        batch_shape = _mul_broadcast_shape(*[lt.shape for lt in lazy_tensors])
        lazy_tensors = tuple(lt.expand(batch_shape) for lt in lazy_tensors)
        super(SumLazyTensor, self).__init__(*lazy_tensors, **kwargs)

        self.lazy_tensors = lazy_tensors

Instances