
    bIg                    N    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	y)    )annotationsN)Base)expectc                  0    e Zd Zedd       Zedd       Zy)Shrinkc                 
   t         j                  j                  ddgdgd      } t        j                  ddt        j
                        }t        j                  g d	t        j
                        }t        | |g|gd
       y )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      _/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/shrink.pyexport_hard_shrinkzShrink.export_hard_shrink   sk    {{$$5E	 % 
 IIdCrzz2HH%RZZ8tQC!3EF    c                    t         j                  j                  ddgdgdd      } t        j                  ddt        j
                        }t        j                  g d	t        j
                        }t        | |g|gd
       y )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   sn    {{$$5E % 
 IIdCrzz2HH)<tQC!3EFr&   N)returnNone)__name__
__module____qualname__staticmethodr%   r*    r&   r$   r   r      s,    	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T Gr&   