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