def create_lazy_tensor(self):
mat1 = make_random_mat(6, rank=6, batch_shape=torch.Size((
2,
3,
)))
mat2 = make_random_mat(6, rank=6, batch_shape=torch.Size((
2,
3,
)))
res = RootLazyTensor(mat1) * RootLazyTensor(mat2)
return res.add_diag(torch.tensor(0.5))
def create_lazy_tensor(self):
mat1 = make_random_mat(6, 6)
mat2 = make_random_mat(6, 6)
res = RootLazyTensor(mat1) * RootLazyTensor(mat2)
return res.add_diag(torch.tensor(2.0))
