U
    rh                      @  sH   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 deZ	dS )    )annotationsN)Base)expectc                   @  s0   e Zd ZeddddZeddddZdS )SeluNone)returnc                  C  s   t jjddgdgddd} tddd	gtj}t|dtjd t	t|tj dd	 d d  }t
| |g|gd
d tjdddtj}t|dtjd t	t|tj dd	 d d  }t
| |g|gdd d S )Nr   xyg       @g      @)inputsoutputsalphagammar      Ztest_selu_exampler
   r   name         Z	test_selu)onnxhelper	make_nodenparrayastypefloat32clipinfexpr   randomrandn)noder   r	    r"   D/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/selu.pyexport   s&        ""zSelu.exportc                  C  s   d} d}t jjddgdgd}tjddd	tj}t|d
tj	| t
t|tj	 d
d |  |  }t||g|gdd d S )Ng   ?g   ?r   r   r	   )r
   r   r   r   r   r   r   Ztest_selu_defaultr   )r   r   r   r   r   r    r   r   r   r   r   r   )Zdefault_alphaZdefault_gammar!   r   r	   r"   r"   r#   export_selu_default#   s    "zSelu.export_selu_defaultN)__name__
__module____qualname__staticmethodr$   r%   r"   r"   r"   r#   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r"   r"   r"   r#   <module>   s
   