o
    wi                     @  s>   d dl mZ d dlZd dlmZ ddd	ZG d
d deZdS )    )annotationsN)OpRundata
np.ndarrayshapereturnc                 C  s   t j|| jd}| | S )N)dtype)nponesr   )r   r   r
    r   Y/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_expand.pycommon_reference_implementation   s   r   c                   @  s   e Zd Zdd ZdS )Expandc                 C  s   t ||fS )N)r   )selfr   r   r   r   r   _run   s   zExpand._runN)__name__
__module____qualname__r   r   r   r   r   r      s    r   )r   r   r   r   r   r   )
__future__r   numpyr	   onnx.reference.op_runr   r   r   r   r   r   r   <module>   s
   
