edwin-zft

When I use opt_einsum optimizes torch.einum, the running time after optimization increases.

5

``` import numpy as np import time import torch from opt_einsum import contract dim = 4 x = torch.randn(6 ,4 ,4, 4) w1 = torch.randn(1,4,dim) w2 = torch.randn(dim,4,dim) w3 =...