def fun(x): return rewrites.constant_folding_einsum(',i,i,j->', np.sum(x), np.ones_like(x), x, ones)
def fun(x): return rewrites.constant_folding_einsum(',j,i,j->', np.sum(x), 0. * ones, x, 2.0 * ones)
def fun(x): return rewrites.constant_folding_einsum(',,i,->', np.sum(x), 0., x, 2.0)