@@ -18,12 +18,12 @@ def setup(self):
18
18
self .a3 = np .arange (480000. ).reshape (60 , 80 , 100 )
19
19
self .b3 = np .arange (192000. ).reshape (80 , 60 , 40 )
20
20
21
- def time_einsum_ij_jk_a_b (self ):
22
- np .einsum ('ij,jk' , self .a , self .b )
23
-
24
21
def time_dot_a_b (self ):
25
22
np .dot (self .a , self .b )
26
23
24
+ def time_dot_d_dot_b_c (self ):
25
+ np .dot (self .d , np .dot (self .b , self .c ))
26
+
27
27
def time_dot_trans_a_at (self ):
28
28
np .dot (self .a , self .at )
29
29
@@ -36,20 +36,38 @@ def time_dot_trans_at_a(self):
36
36
def time_dot_trans_atc_a (self ):
37
37
np .dot (self .atc , self .a )
38
38
39
+ def time_einsum_i_ij_j (self ):
40
+ np .einsum ('i,ij,j' , self .d , self .b , self .c )
41
+
42
+ def time_einsum_ij_jk_a_b (self ):
43
+ np .einsum ('ij,jk' , self .a , self .b )
44
+
45
+ def time_einsum_ijk_jil_kl (self ):
46
+ np .einsum ('ijk,jil->kl' , self .a3 , self .b3 )
47
+
39
48
def time_inner_trans_a_a (self ):
40
49
np .inner (self .a , self .a )
41
50
42
51
def time_inner_trans_a_ac (self ):
43
52
np .inner (self .a , self .ac )
44
53
45
- def time_einsum_i_ij_j (self ):
46
- np .einsum ( 'i,ij,j' , self .d , self .b , self . c )
54
+ def time_matmul_a_b (self ):
55
+ np .matmul ( self .a , self .b )
47
56
48
- def time_dot_d_dot_b_c (self ):
49
- np .dot (self .d , np .dot (self .b , self .c ))
57
+ def time_matmul_d_matmul_b_c (self ):
58
+ np .matmul (self .d , np .matmul (self .b , self .c ))
50
59
51
- def time_einsum_ijk_jil_kl (self ):
52
- np .einsum ('ijk,jil->kl' , self .a3 , self .b3 )
60
+ def time_matmul_trans_a_at (self ):
61
+ np .matmul (self .a , self .at )
62
+
63
+ def time_matmul_trans_a_atc (self ):
64
+ np .matmul (self .a , self .atc )
65
+
66
+ def time_matmul_trans_at_a (self ):
67
+ np .matmul (self .at , self .a )
68
+
69
+ def time_matmul_trans_atc_a (self ):
70
+ np .matmul (self .atc , self .a )
53
71
54
72
def time_tensordot_a_b_axes_1_0_0_1 (self ):
55
73
np .tensordot (self .a3 , self .b3 , axes = ([1 , 0 ], [0 , 1 ]))
0 commit comments