Coverage for mlprodict/onnx_conv/onnx_ops/onnx_complex.py: 100%

1""" 

2@file 

3@brief Custom operators for complex numbers. 

4""" 

5from skl2onnx.algebra.onnx_operator import OnnxOperator 

6 

7 

8class OnnxComplexAbs_1(OnnxOperator): 

9 """ 

10 Defines a custom operator for ComplexAbs. 

11 """ 

12 

13 since_version = 1 

14 expected_inputs = [('X', 'T'), ] 

15 expected_outputs = [('Y', 'U')] 

16 input_range = [1, 1] 

17 output_range = [1, 1] 

18 is_deprecated = False 

19 domain = 'mlprodict' 

20 operator_name = 'ComplexAbs' 

21 past_version = {} 

22 

23 def __init__(self, X, op_version=None, **kwargs): 

24 """ 

25 :param X: array or OnnxOperatorMixin 

26 :param op_version: opset version 

27 :param kwargs: additional parameter 

28 """ 

29 OnnxOperator.__init__( 

30 self, X, op_version=op_version, **kwargs) 

31 

32 

33OnnxComplexAbs = OnnxComplexAbs_1