
    bIg                    :    d dl mZ d dlZd dlmZ  G d de      Zy)    )annotationsN)
OpRunUnaryc                      e Zd ZddZy)FlattenNc                    |xs | j                   }|j                  }|dk(  rdn,t        j                  |d |       j	                  t
              df}|j                  |      fS )Nr   )   r	   )axisshapenpprodastypeintreshape)selfxr
   ir   	new_shapes         Z/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/reference/ops/op_flatten.py_runzFlatten._run   sW    DII AvGBGGE"1I,>,E,Ec,JB+O			)$&&    )N)__name__
__module____qualname__r    r   r   r   r      s    'r   r   )
__future__r   numpyr   onnx.reference.ops._opr   r   r   r   r   <module>r      s    #  -'j 'r   