o
    toi                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)OpRunc                   @  s   e Zd ZddddZdS )ConstantOfShapeNvaluenp.array | Nonec                 C  s   | j d u rtjdtjd}n| j jdkr$td| jjdt| j  d| j j	d }zt
t||}W |fS  tyR } ztd|d|d	| j d
|d }~ww )Nr   )dtype   zOperator ConstantOfShape (zG) expects a single element tensor as value, but the size of 'value' is .z%Unable to create a constant of shape z with value z (raw value=z).)r   nparrayfloat32size
ValueError	onnx_nodenamelenflatfulltuple	TypeErrorRuntimeError)selfdatar   rese r   [/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_constant_of_shape.py_run   s(   
zConstantOfShape._run)N)r   r   )__name__
__module____qualname__r   r   r   r   r   r      s    r   )
__future__r   numpyr
   onnx.reference.op_runr   r   r   r   r   r   <module>   s   