U
    qh                     @  sF   d dl mZ d dlZd dlmZ ddddddZG dd	 d	eZdS )
    )annotationsN)OpRunz
np.ndarray)datashapereturnc                 C  s   t j|| jd}| | S )N)dtype)nponesr   )r   r   r	    r
   @/tmp/pip-unpacked-wheel-xnis5xre/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   )
__future__r   Znumpyr   Zonnx.reference.op_runr   r   r   r
   r
   r
   r   <module>   s   