Coverage for mlprodict/testing/einsum/__init__.py: 100%

1""" 

2@file 

3@brief Shortcut to *testing.einsum*. 

4""" 

5 

6from .einsum_bench import einsum_benchmark 

7from .einsum_fct import einsum, optimize_decompose_einsum_equation 

8from .einsum_impl import decompose_einsum_equation, apply_einsum_sequence 

9from .einsum_impl_classes import EinsumSubOp, GraphEinsumSubOp 

10from .einsum_impl_ext import ( 

11 numpy_extended_dot, 

12 numpy_extended_dot_python, 

13 numpy_extended_dot_matrix, 

14 numpy_diagonal)