
    /jW                    V    d dl mZ d dlZd dlZd dlmZ d dlmZ  G d de          Z	dS )    )annotationsN)Base)expectc                  &    e Zd Zedd            ZdS )MeanreturnNonec                 N   t          j        g d                              t           j                  } t          j        g d                              t           j                  }t          j        g d                              t           j                  }t          j        g d                              t           j                  }t          j                            dg ddg          }t          || ||g|gd	
           t          j                            ddgdg          }t          || g| gd
           t          j        t          j	        | |          d          }t          j                            dddgdg          }t          || |g|gd
           d S )N)   r      )   r      )r      r   )r   r   r   r   )data_0data_1data_2result)inputsoutputstest_mean_example)r   r   namer   test_mean_one_inputg       @r   test_mean_two_inputs)
nparrayastypefloat32onnxhelper	make_noder   divideadd)r   r   r   r   nodes        e/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/mean.pyexportzMean.export   s   )))$$++BJ77)))$$++BJ77)))$$++BJ77)))$$++BJ77{$$111J % 
 

 	FF+H$		
 	
 	
 	
 {$$:J % 
 

 	tVHvh=RSSSS26&&11377{$$h'J % 
 

 	&&)F8BX	
 	
 	
 	
 	
 	
    N)r   r	   )__name__
__module____qualname__staticmethodr%    r&   r$   r   r      s2         
  
  
 \ 
  
  
r&   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r+   r&   r$   <module>r0      s    # " " " " "      , , , , , , . . . . . ."
 "
 "
 "
 "
4 "
 "
 "
 "
 "
r&   