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                   @  s   e Zd ZeddddZdS )ConstantNone)returnc                  C  sb   t jddt j} tjjdg dgtjjdtj	j
| j|  tdd}t|g | gdd d S )	N   r   valuesZconst_tensor)nameZ	data_typeZdimsvals)inputsoutputsvalueZtest_constant)r   r   r
   )nprandomZrandnZastypeZfloat32onnxhelperZ	make_nodeZmake_tensorZTensorProtoFLOATshapeflattenfloatr   )r	   node r   H/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/constant.pyexport   s    zConstant.exportN)__name__
__module____qualname__staticmethodr   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
   