
    /j                    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edd            ZdS )ShrinkreturnNonec                     t           j                            ddgdgd          } t          j        ddt          j                  }t          j        g d	t          j                  }t          | |g|gd
           d S )Nr   xy      ?)inputsoutputslambd        @dtype)r   r   r      test_shrink_hardr   r   nameonnxhelper	make_nodenparangefloat32arrayr   nodeXYs      g/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/shrink.pyexport_hard_shrinkzShrink.export_hard_shrink   s    {$$5E	 % 
 
 IdCrz222H%%%RZ888tQC!3EFFFFFF    c                    t           j                            ddgdgdd          } t          j        ddt          j                  }t          j        g d	t          j                  }t          | |g|gd
           d S )Nr   r   r   r   )r   r   r   biasr   r   r   )g      r   r   r   g      ?test_shrink_softr   r   r"   s      r&   export_soft_shrinkzShrink.export_soft_shrink   s    {$$5E % 
 
 IdCrz222H)))<<<tQC!3EFFFFFFr(   N)r   r	   )__name__
__module____qualname__staticmethodr'   r,    r(   r&   r   r      s\        	G 	G 	G \	G 
G 
G 
G \
G 
G 
Gr(   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r1   r(   r&   <module>r6      s    # " " " " "      , , , , , , . . . . . .G G G G GT G G G G Gr(   