U
    q±Ëh  ã                   @  sD   d dl mZ d dlZd dlmZ dddœdd„ZG dd	„ d	eƒZdS )
é    )ÚannotationsN)ÚOpRunz
np.ndarray)ÚxÚreturnc                 C  s<   t tdt | ¡ƒƒ}tj| |d}|D ]}t |d¡}q&|S )Né   )Úaxiséÿÿÿÿ)ÚtupleÚrangeÚnpÚndimZaverageZexpand_dims)r   r   ÚyÚ_© r   úM/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_global_average_pool.pyÚ_global_average_pool   s
    r   c                   @  s   e Zd Zdd„ ZdS )ÚGlobalAveragePoolc                 C  s   t |ƒ |j¡fS )N)r   ZastypeZdtype)Úselfr   r   r   r   Ú_run   s    zGlobalAveragePool._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   