U
    qh+                     @  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  sF   t | d }| jtt||d d}t|D ]}t |d}q0|S )N   )Zaxis)npndimmaxtuplerangeZexpand_dims)r   Zspatial_shapey_ r   I/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_global_max_pool.py_global_max_pool   s
    r   c                   @  s   e Zd Zdd ZdS )GlobalMaxPoolc                 C  s   t |}|fS )N)r   )selfr   resr   r   r   _run   s    zGlobalMaxPool._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   