
    
Ng                     6    d dl Zd dlmZ  G d de          ZdS )    N)
OpRunUnaryc                       e Zd ZddZdS )FlattenNc                     |p| j         }|j        }|dk    rdn5t          j        |d |                                       t
                    df}|                    |          fS )Nr   )   r   )axisshapenpprodastypeintreshape)selfxr	   ir
   	new_shapes         Y/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/reference/ops/op_flatten.py_runzFlatten._run   s`    DI AvvGGBGE"1"I,>,>,E,Ec,J,JB+O			)$$&&    )N)__name__
__module____qualname__r    r   r   r   r      s(        ' ' ' ' ' 'r   r   )numpyr   onnx.reference.ops._opr   r   r   r   r   <module>r      sY        - - - - - -' ' ' ' 'j ' ' ' ' 'r   