mul_lazy_variable.py 文件源码

def add_diag(self, diag):
        if self.added_diag is None:
            return MulLazyVariable(*self.lazy_vars,
                                   matmul_mode=self.matmul_mode,
                                   max_iter=self.max_iter,
                                   num_samples=self.num_samples,
                                   added_diag=diag.expand(self.size()[0]))
        else:
            return MulLazyVariable(*self.lazy_vars,
                                   matmul_mode=self.matmul_mode,
                                   max_iter=self.max_iter,
                                   num_samples=self.num_samples,
                                   added_diag=self.added_diag + diag)