## matrix inversion
import time
import torch

n = 10000
r = torch.rand(n, n)
i = torch.rand(n, n)
c = torch.complex(r, i)
#
start_time = time.time()
cinv_cpu = c.inverse()
print(f"CPU time : {time.time() - start_time} seconds")
#
d = c.cuda()
start_time = time.time()
dinv_cpu = d.inverse()
print(f"GPU time : {time.time() - start_time} seconds")